[ci skip]

Rejoker456 · Sep 8, 2021 · 3f6db83 · 3f6db83
1 parent b11e3b9
commit 3f6db83
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -49,6 +49,8 @@ def get_extensions():
             nvcc_flags = os.getenv('NVCC_FLAGS', '')
             nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
             nvcc_flags += ['--expt-relaxed-constexpr', '-O2']
+            if sys.platform == 'win32':
+                nvcc_flags += ['/Zm200']
             extra_compile_args['nvcc'] = nvcc_flags
 
         name = main.split(os.sep)[-1][:-4]