File tree 1 file changed +2
-6
lines changed 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -181,7 +181,7 @@ def get_extensions():
181
181
182
182
define_macros = []
183
183
184
- extra_compile_args = {}
184
+ extra_compile_args = {'cxx' : [] }
185
185
if (torch .cuda .is_available () and ((CUDA_HOME is not None ) or is_rocm_pytorch )) \
186
186
or os .getenv ('FORCE_CUDA' , '0' ) == '1' :
187
187
extension = CUDAExtension
@@ -196,15 +196,11 @@ def get_extensions():
196
196
else :
197
197
define_macros += [('WITH_HIP' , None )]
198
198
nvcc_flags = []
199
- extra_compile_args = {
200
- 'cxx' : [],
201
- 'nvcc' : nvcc_flags ,
202
- }
199
+ extra_compile_args ["nvcc" ] = nvcc_flags
203
200
204
201
if sys .platform == 'win32' :
205
202
define_macros += [('torchvision_EXPORTS' , None )]
206
203
207
- extra_compile_args .setdefault ('cxx' , [])
208
204
extra_compile_args ['cxx' ].append ('/MP' )
209
205
210
206
debug_mode = os .getenv ('DEBUG' , '0' ) == '1'
You can’t perform that action at this time.
0 commit comments