Disable backward for now, we need to debug the launch templates.

Author: LoserCheems

setup.py

@@ -269,7 +269,8 @@ def append_nvcc_threads(nvcc_extra_args):
                         # "--ptxas-options=-v",
                         # "--ptxas-options=-O2",
                         # "-lineinfo",
-                        # "-DFLASHATTENTION_DISABLE_BACKWARD",  # Only forward pass
+                        # Disable backward for now, we need to debug the launch templates.
+                        "-DFLASHATTENTION_DISABLE_BACKWARD",  # Only forward pass
                         "-DFLASHATTENTION_DISABLE_SOFTCAP",
                         "-DFLASHATTENTION_DISABLE_UNEVEN_K",
                     ]
@@ -387,4 +388,4 @@ def __init__(self, *args, **kwargs) -> None:
     },
     # Ensure the package is properly included
     include_package_data=True,
-)
+)