Remove dropout & Uneven K

Author: WoosukKwon

setup.py

@@ -201,9 +201,9 @@ def append_nvcc_threads(nvcc_extra_args):
                         # "--ptxas-options=-O2",
                         # "-lineinfo",
                         # "-DFLASHATTENTION_DISABLE_BACKWARD",
-                        # "-DFLASHATTENTION_DISABLE_DROPOUT",
+                        "-DFLASHATTENTION_DISABLE_DROPOUT",
                         # "-DFLASHATTENTION_DISABLE_ALIBI",
-                        # "-DFLASHATTENTION_DISABLE_UNEVEN_K",
+                        "-DFLASHATTENTION_DISABLE_UNEVEN_K",
                         # "-DFLASHATTENTION_DISABLE_LOCAL",
                     ]
                     + generator_flag