summarylogtreecommitdiffstats
path: root/flash-attention.diff
blob: 2bb3a3402dd3a3b9cfcdd33a2d78f973c1f44cfa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
--- setup.py	2023-12-24 11:32:47.740182920 +0300
+++ setup.py	2023-12-24 11:34:31.217200306 +0300
@@ -89,10 +89,6 @@
 cmdclass = {}
 ext_modules = []
 
-# We want this even if SKIP_CUDA_BUILD because when we run python setup.py sdist we want the .hpp
-# files included in the source distribution, in case the user compiles from source.
-subprocess.run(["git", "submodule", "update", "--init", "csrc/cutlass"])
-
 if not SKIP_CUDA_BUILD:
     print("\n\ntorch.__version__  = {}\n\n".format(torch.__version__))
     TORCH_MAJOR = int(torch.__version__.split(".")[0])
@@ -207,7 +203,6 @@
             include_dirs=[
                 Path(this_dir) / "csrc" / "flash_attn",
                 Path(this_dir) / "csrc" / "flash_attn" / "src",
-                Path(this_dir) / "csrc" / "cutlass" / "include",
             ],
         )
     )
@@ -319,6 +314,5 @@
         "torch",
         "einops",
         "packaging",
-        "ninja",
     ],
 )