Add -allow-unsupported-compiler to nvcc compiler for windows (#27)

casper-hansen · Jul 23, 2024 · 9f56083 · 9f56083
1 parent 3da9d7f
commit 9f56083
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -118,7 +118,7 @@ def get_extra_compile_args(arch_flags, generator_flags):
         include_arch = os.getenv("INCLUDE_ARCH", "1") == "1"
         # Relaxed args on Windows
         if include_arch:
-            extra_compile_args = {"nvcc": arch_flags}
+            extra_compile_args = {"nvcc": ["-allow-unsupported-compiler"] + arch_flags}
 
     elif CUDA_VERSION:
         extra_compile_args = {