1 parent 6fab369 commit 84657d3Copy full SHA for 84657d3
1 file changed
setup.py
@@ -74,7 +74,7 @@ def get_extensions():
74
nvcc_flags += ['-O2']
75
nvcc_flags += ['-DTORCH_INDUCTOR_CPP_WRAPPER']
76
if sys.platform == 'win32':
77
- nvcc_flags += ['-D_WIN32=1']
+ nvcc_flags += ['-Xcompiler', '/permissive-']
78
extra_compile_args['nvcc'] = nvcc_flags
79
80
if torch.version.hip:
0 commit comments