Allow local cutlass directory to be used in sgl-kernel build (#3037)

2025-01-24 03:59:47 -08:00
parent 153b414e83
commit 685a5738a7
1 changed files with 2 additions and 0 deletions
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -39,6 +39,8 @@ def _get_version():


 cutlass = root / "3rdparty" / "cutlass"
+cutlass_default = root / "3rdparty" / "cutlass"
+cutlass = Path(os.environ.get("CUSTOM_CUTLASS_SRC_DIR", default=cutlass_default))
 flashinfer = root / "3rdparty" / "flashinfer"
 include_dirs = [
    cutlass.resolve() / "include",