Skip to content

GitLab

Commit 839dd6a6 authored by rusty1s's avatar rusty1s
Browse files

reduce shared library file size

parent 46dac04f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
setup.py
View file @ 839dd6a6
...@@ -35,8 +35,8 @@ def get_extensions(): ...@@ -35,8 +35,8 @@ def get_extensions():
define_macros += [('MTMETIS_64BIT_WEIGHTS', None)] define_macros += [('MTMETIS_64BIT_WEIGHTS', None)]
define_macros += [('MTMETIS_64BIT_PARTITIONS', None)] define_macros += [('MTMETIS_64BIT_PARTITIONS', None)]
libraries += ['mtmetis', 'wildriver'] libraries += ['mtmetis', 'wildriver']
extra_compile_args = {'cxx': []} extra_compile_args = {'cxx': ['-O2']}
extra_link_args = [] extra_link_args = ['-s']
info = parallel_info() info = parallel_info()
if 'parallel backend: OpenMP' in info and 'OpenMP not found' not in info: if 'parallel backend: OpenMP' in info and 'OpenMP not found' not in info:
...@@ -53,7 +53,7 @@ def get_extensions(): ...@@ -53,7 +53,7 @@ def get_extensions():
define_macros += [('WITH_CUDA', None)] define_macros += [('WITH_CUDA', None)]
nvcc_flags = os.getenv('NVCC_FLAGS', '') nvcc_flags = os.getenv('NVCC_FLAGS', '')
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ') nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr'] nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr', '-O2']
extra_compile_args['nvcc'] = nvcc_flags extra_compile_args['nvcc'] = nvcc_flags
if sys.platform == 'win32': if sys.platform == 'win32':
......
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