Skip to content

GitLab

Commit 03e02e6c authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent 26823fc5
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
setup.py
View file @ 03e02e6c
...@@ -33,6 +33,9 @@ def get_extensions(): ...@@ -33,6 +33,9 @@ def get_extensions():
extra_compile_args = {'cxx': ['-O2']} extra_compile_args = {'cxx': ['-O2']}
extra_link_args = ['-s'] extra_link_args = ['-s']
if sys.platform == 'win32':
extra_compile_args['cxx'].append('/Zm500')
info = parallel_info() info = parallel_info()
if ('backend: OpenMP' in info and 'OpenMP not found' not in info if ('backend: OpenMP' in info and 'OpenMP not found' not in info
and sys.platform != 'darwin'): and sys.platform != 'darwin'):
...@@ -50,6 +53,8 @@ def get_extensions(): ...@@ -50,6 +53,8 @@ def get_extensions():
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ') nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
nvcc_flags += ['--expt-relaxed-constexpr', '-O2'] nvcc_flags += ['--expt-relaxed-constexpr', '-O2']
extra_compile_args['nvcc'] = nvcc_flags extra_compile_args['nvcc'] = nvcc_flags
if sys.platform == 'win32':
extra_compile_args['nvcc'].append('/Zm500')
name = main.split(os.sep)[-1][:-4] name = main.split(os.sep)[-1][:-4]
sources = [main] sources = [main]
......
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