Skip to content

GitLab

Unverified Commit 685a5738 authored by Trevor Morris's avatar Trevor Morris Committed by GitHub
Browse files

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

parent 153b414e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
sgl-kernel/setup.py
View file @ 685a5738
......@@ -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",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment