remove extra_compile_args for support windows

b621c142 · qwopqwop200 · GitHub · 81fd57c3 · b621c142
Unverified Commit b621c142 authored Sep 21, 2023 by qwopqwop200 Committed by GitHub Sep 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

setup.py setup.py +2 -4

No files found.
--- a/setup.py
+++ b/setup.py
@@ -97,9 +97,7 @@ arch_flags = get_compute_capabilities()

 if os.name == "nt":
    # Relaxed args on Windows
-    extra_compile_args={
-        "nvcc": arch_flags
-    }
+    extra_compile_args={}
 else:
    extra_compile_args={
        "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17", "-DENABLE_BF16"],
@@ -156,4 +154,4 @@ setup(
    install_requires=requirements,
    include_dirs=include_dirs,
    **common_setup_kwargs
-)
\ No newline at end of file
+)