Skip to content

GitLab

Unverified Commit b621c142 authored by qwopqwop200's avatar qwopqwop200 Committed by GitHub
Browse files

remove extra_compile_args for support windows

parent 81fd57c3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
setup.py
View file @ b621c142
...@@ -97,9 +97,7 @@ arch_flags = get_compute_capabilities() ...@@ -97,9 +97,7 @@ arch_flags = get_compute_capabilities()
if os.name == "nt": if os.name == "nt":
# Relaxed args on Windows # Relaxed args on Windows
extra_compile_args={ extra_compile_args={}
"nvcc": arch_flags
}
else: else:
extra_compile_args={ extra_compile_args={
"cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17", "-DENABLE_BF16"], "cxx": ["-g", "-O3", "-fopenmp", "-lgomp", "-std=c++17", "-DENABLE_BF16"],
...@@ -156,4 +154,4 @@ setup( ...@@ -156,4 +154,4 @@ setup(
install_requires=requirements, install_requires=requirements,
include_dirs=include_dirs, include_dirs=include_dirs,
**common_setup_kwargs **common_setup_kwargs
) )
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment