potential fix

26c0b866 · rusty1s · b2cddcbd · 26c0b866
Commit 26c0b866 authored Feb 29, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

setup.py setup.py +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -26,6 +26,7 @@ def get_extensions():
        define_macros += [('WITH_CUDA', None)]
        nvcc_flags = os.getenv('NVCC_FLAGS', '')
        nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
+        nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr']
        extra_compile_args['nvcc'] = nvcc_flags

    extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc')