From 5dfcacfcb186ca7d35ce535d0ab3c34df6eff7ea Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Tue, 21 Jan 2025 00:04:12 +0800 Subject: [PATCH] Add compile flags for cutlass 3.x (#3013) Co-authored-by: HandH1998 <1335248067@qq.com> --- sgl-kernel/setup.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index 25319af7a..9f9867113 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -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__", ]