initial include dirs

alexzhang13 · alexzhang13 · commit cec38a75158d · 2025-01-14T12:04:30.000-05:00
diff --git a/src/discord-cluster-manager/consts.py b/src/discord-cluster-manager/consts.py
@@ -68,4 +68,4 @@ def combine_enums(enums: list[Type[Enum]], combined_name: str) -> Enum:
     "-Xptxas=--verbose",
     "-Xptxas=--warn-on-spills",
 ]
-MODAL_CUDA_INCLUDE_DIRS = ["-I/ThunderKittens/include"]
+MODAL_CUDA_INCLUDE_DIRS = ["-I/ThunderKittens/include", "-I/cutlass/include"]
diff --git a/src/discord-cluster-manager/modal_runner.py b/src/discord-cluster-manager/modal_runner.py
@@ -42,6 +42,8 @@
         #
         "git clone https://github.com/HazyResearch/ThunderKittens.git",
         # "cd /ThunderKittens && pwd && python setup.py install",
+        # CUTLASS
+        "git clone https://github.com/NVIDIA/cutlass.git",
     )
 )
 

Original file line number	Diff line number	Diff line change
`@@ -68,4 +68,4 @@ def combine_enums(enums: list[Type[Enum]], combined_name: str) -> Enum:`
`68`	`68`	`"-Xptxas=--verbose",`
`69`	`69`	`"-Xptxas=--warn-on-spills",`
`70`	`70`	`]`
`71`		`-MODAL_CUDA_INCLUDE_DIRS = ["-I/ThunderKittens/include"]`
	`71`	`+MODAL_CUDA_INCLUDE_DIRS = ["-I/ThunderKittens/include", "-I/cutlass/include"]`
Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,8 @@`
`42`	`42`	`#`
`43`	`43`	`"git clone https://github.com/HazyResearch/ThunderKittens.git",`
`44`	`44`	`# "cd /ThunderKittens && pwd && python setup.py install",`
	`45`	`+ # CUTLASS`
	`46`	`+ "git clone https://github.com/NVIDIA/cutlass.git",`
`45`	`47`	`)`
`46`	`48`	`)`
`47`	`49`