Commit 14d4f8cb authored by Casper's avatar Casper
Browse files

Include arch by default, disable with INCLUDE_ARCH=0

parent b621c142
...@@ -96,7 +96,12 @@ generator_flags = get_generator_flag() ...@@ -96,7 +96,12 @@ generator_flags = get_generator_flag()
arch_flags = get_compute_capabilities() arch_flags = get_compute_capabilities()
if os.name == "nt": if os.name == "nt":
include_arch = os.getenv("INCLUDE_ARCH", "1") == "1"
# Relaxed args on Windows # Relaxed args on Windows
if include_arch:
extra_compile_args={"nvcc": arch_flags}
else:
extra_compile_args={} extra_compile_args={}
else: else:
extra_compile_args={ extra_compile_args={
......
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