1 parent 00e4a5a commit 6fab369Copy full SHA for 6fab369
1 file changed
setup.py
@@ -73,6 +73,8 @@ def get_extensions():
73
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
74
nvcc_flags += ['-O2']
75
nvcc_flags += ['-DTORCH_INDUCTOR_CPP_WRAPPER']
76
+ if sys.platform == 'win32':
77
+ nvcc_flags += ['-D_WIN32=1']
78
extra_compile_args['nvcc'] = nvcc_flags
79
80
if torch.version.hip:
0 commit comments