Skip to content

Commit 57dc11e

Browse files
atuponeNicolasHug
andauthored
split NVCC_FLAGS but not quoted string (#8933)
Co-authored-by: Nicolas Hug <[email protected]>
1 parent aeed08e commit 57dc11e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import subprocess
77
import sys
88
import warnings
9+
import shlex
910
from pathlib import Path
1011

1112
import torch
@@ -123,7 +124,7 @@ def get_macros_and_flags():
123124
if NVCC_FLAGS is None:
124125
nvcc_flags = []
125126
else:
126-
nvcc_flags = NVCC_FLAGS.split(" ")
127+
nvcc_flags = shlex.split(NVCC_FLAGS)
127128
extra_compile_args["nvcc"] = nvcc_flags
128129

129130
if sys.platform == "win32":

0 commit comments

Comments
 (0)