diff --git a/src/discord-cluster-manager/consts.py b/src/discord-cluster-manager/consts.py index 5f026971..fb012907 100644 --- 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 index 97289467..a8226711 100644 --- 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", ) )