Skip to content

Commit f319a87

Browse files
committed
[ci skip]
1 parent 77f6ba5 commit f319a87

File tree

2 files changed

+1
-4
lines changed

2 files changed

+1
-4
lines changed

.github/workflows/cuda/cu111-Windows-env.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
55
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
66

77
export FORCE_CUDA=1
8-
export TORCH_CUDA_ARCH_LIST="5.0+PTX;6.0;7.0;7.5;8.0;8.6"
8+
export TORCH_CUDA_ARCH_LIST="6.0+PTX;7.0;7.5;8.0;8.6"

setup.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,6 @@ def get_extensions():
3333
extra_compile_args = {'cxx': ['-O2']}
3434
extra_link_args = ['-s']
3535

36-
if sys.platform == 'win32':
37-
extra_compile_args['cxx'].append('/Zm500')
38-
3936
info = parallel_info()
4037
if ('backend: OpenMP' in info and 'OpenMP not found' not in info
4138
and sys.platform != 'darwin'):

0 commit comments

Comments
 (0)