File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -35,8 +35,8 @@ def get_extensions():
3535 define_macros += [('MTMETIS_64BIT_WEIGHTS' , None )]
3636 define_macros += [('MTMETIS_64BIT_PARTITIONS' , None )]
3737 libraries += ['mtmetis' , 'wildriver' ]
38- extra_compile_args = {'cxx' : []}
39- extra_link_args = []
38+ extra_compile_args = {'cxx' : ['-O2' ]}
39+ extra_link_args = ['-s' ]
4040
4141 info = parallel_info ()
4242 if 'parallel backend: OpenMP' in info and 'OpenMP not found' not in info :
@@ -53,7 +53,7 @@ def get_extensions():
5353 define_macros += [('WITH_CUDA' , None )]
5454 nvcc_flags = os .getenv ('NVCC_FLAGS' , '' )
5555 nvcc_flags = [] if nvcc_flags == '' else nvcc_flags .split (' ' )
56- nvcc_flags += ['-arch=sm_35' , '--expt-relaxed-constexpr' ]
56+ nvcc_flags += ['-arch=sm_35' , '--expt-relaxed-constexpr' , '-O2' ]
5757 extra_compile_args ['nvcc' ] = nvcc_flags
5858
5959 if sys .platform == 'win32' :
You can’t perform that action at this time.
0 commit comments