Skip to content

Commit 6fab369

Browse files
committed
update
1 parent 00e4a5a commit 6fab369

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,8 @@ def get_extensions():
7373
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
7474
nvcc_flags += ['-O2']
7575
nvcc_flags += ['-DTORCH_INDUCTOR_CPP_WRAPPER']
76+
if sys.platform == 'win32':
77+
nvcc_flags += ['-D_WIN32=1']
7678
extra_compile_args['nvcc'] = nvcc_flags
7779

7880
if torch.version.hip:

0 commit comments

Comments
 (0)