Skip to content

Commit 1cf7fd4

Browse files
committed
Removes commented dropout disable flag
Cleans up build configuration by removing unused commented compilation flag for disabling dropout functionality
1 parent 35ac6d8 commit 1cf7fd4

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

setup.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,6 @@ def append_nvcc_threads(nvcc_extra_args):
244244
# "--ptxas-options=-O2",
245245
# "-lineinfo",
246246
"-DFLASHATTENTION_DISABLE_BACKWARD", # Only forward pass
247-
# "-DFLASHATTENTION_DISABLE_DROPOUT",
248247
# "-DFLASHATTENTION_DISABLE_SOFTCAP",
249248
# "-DFLASHATTENTION_DISABLE_UNEVEN_K",
250249
]

0 commit comments

Comments
 (0)