From cec38a75158defad52b557f02383ff201639e67a Mon Sep 17 00:00:00 2001 From: Alex Zhang Date: Tue, 14 Jan 2025 12:04:30 -0500 Subject: [PATCH] initial include dirs --- src/discord-cluster-manager/consts.py | 2 +- src/discord-cluster-manager/modal_runner.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/discord-cluster-manager/consts.py b/src/discord-cluster-manager/consts.py index 5f0269717..fb012907e 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 97289467c..a82267111 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", ) )