From a28a6055f947816e455e6a723ed967a8649079ef 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 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", ) )