[ci skip]

Author: rusty1s

setup.py

@@ -33,6 +33,9 @@ def get_extensions():
         extra_compile_args = {'cxx': ['-O2']}
         extra_link_args = ['-s']
 
+        if sys.platform == 'win32':
+            extra_compile_args['cxx'].append('/Zm500')
+
         info = parallel_info()
         if ('backend: OpenMP' in info and 'OpenMP not found' not in info
                 and sys.platform != 'darwin'):
@@ -50,6 +53,8 @@ def get_extensions():
             nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
             nvcc_flags += ['--expt-relaxed-constexpr', '-O2']
             extra_compile_args['nvcc'] = nvcc_flags
+            if sys.platform == 'win32':
+                extra_compile_args['nvcc'].append('/Zm500')
 
         name = main.split(os.sep)[-1][:-4]
         sources = [main]