"vscode:/vscode.git/clone" did not exist on "2925df7cdf349f51dbda463d59569067e8a87c43"
Unverified Commit 5dfcacfc authored by Ke Bao's avatar Ke Bao Committed by GitHub
Browse files

Add compile flags for cutlass 3.x (#3013)


Co-authored-by: default avatarHandH1998 <1335248067@qq.com>
parent 41a0ccd4
......@@ -30,6 +30,7 @@ include_dirs = [
root / "src" / "sgl-kernel" / "csrc",
]
nvcc_flags = [
"-DNDEBUG",
"-O3",
"-Xcompiler",
"-fPIC",
......@@ -37,6 +38,7 @@ nvcc_flags = [
"-gencode=arch=compute_80,code=sm_80",
"-gencode=arch=compute_89,code=sm_89",
"-gencode=arch=compute_90,code=sm_90",
"-gencode=arch=compute_90a,code=sm_90a",
"-U__CUDA_NO_HALF_OPERATORS__",
"-U__CUDA_NO_HALF2_OPERATORS__",
]
......
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