EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Trevor Morris
2025-01-24 03:59:47 -08:00
committed by GitHub
parent 153b414e83
commit 685a5738a7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgl-kernel/setup.py
View File

@@ -39,6 +39,8 @@ def _get_version():
cutlass = root / "3rdparty" / "cutlass" 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" flashinfer = root / "3rdparty" / "flashinfer"
include_dirs = [ include_dirs = [
cutlass.resolve() / "include", cutlass.resolve() / "include",