misc: update build setup (#2306)

2024-12-02 02:03:49 +08:00
parent 7301a39b13
commit 28bc60dcab
5 changed files with 46 additions and 1 deletions
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -13,6 +13,18 @@ setup(
                "src/sgl-kernel/csrc/warp_reduce.cc",
                "src/sgl-kernel/csrc/warp_reduce_kernel.cu",
            ],
+            extra_compile_args={
+                "nvcc": [
+                    "-O3",
+                    "-Xcompiler",
+                    "-fPIC",
+                    "-gencode=arch=compute_75,code=sm_75",
+                    "-gencode=arch=compute_80,code=sm_80",
+                    "-gencode=arch=compute_89,code=sm_89",
+                    "-gencode=arch=compute_90,code=sm_90",
+                ],
+                "cxx": ["-O3"],
+            },
        )
    ],
    cmdclass={"build_ext": BuildExtension},