Commit f319a874 authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent 77f6ba5d
......@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
export FORCE_CUDA=1
export TORCH_CUDA_ARCH_LIST="5.0+PTX;6.0;7.0;7.5;8.0;8.6"
export TORCH_CUDA_ARCH_LIST="6.0+PTX;7.0;7.5;8.0;8.6"
......@@ -33,9 +33,6 @@ 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'):
......
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