Revert "ci: remove multi-threading in nvcc compile flags (flashinfer-…

…ai#229)" (flashinfer-ai#230) This reverts commit f1da148.
octoml · May 3, 2024 · 11ca502 · 11ca502
1 parent f1da148
commit 11ca502
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -394,6 +394,8 @@ def __init__(self, *args, **kwargs) -> None:
                 "nvcc": [
                     "-O3",
                     "-std=c++17",
+                    "--threads",
+                    "8",
                     "-Xfatbin",
                     "-compress-all",
                 ],