Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "5439e917cacc885c0ac39dda1b8af12258e6e16d"
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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
sgl-kernel/setup.py
View file @ 5dfcacfc
...@@ -30,6 +30,7 @@ include_dirs = [ ...@@ -30,6 +30,7 @@ include_dirs = [
root / "src" / "sgl-kernel" / "csrc", root / "src" / "sgl-kernel" / "csrc",
] ]
nvcc_flags = [ nvcc_flags = [
"-DNDEBUG",
"-O3", "-O3",
"-Xcompiler", "-Xcompiler",
"-fPIC", "-fPIC",
...@@ -37,6 +38,7 @@ nvcc_flags = [ ...@@ -37,6 +38,7 @@ nvcc_flags = [
"-gencode=arch=compute_80,code=sm_80", "-gencode=arch=compute_80,code=sm_80",
"-gencode=arch=compute_89,code=sm_89", "-gencode=arch=compute_89,code=sm_89",
"-gencode=arch=compute_90,code=sm_90", "-gencode=arch=compute_90,code=sm_90",
"-gencode=arch=compute_90a,code=sm_90a",
"-U__CUDA_NO_HALF_OPERATORS__", "-U__CUDA_NO_HALF_OPERATORS__",
"-U__CUDA_NO_HALF2_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