Skip to content

Commit d641e89

Browse files
committed
Fix regex pattern for extracting CUDA toolkit version
Signed-off-by: Eric Shi <[email protected]>
1 parent e68a431 commit d641e89

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

warp/build_dll.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def get_cuda_toolkit_version(cuda_home) -> tuple[int, int]:
131131
# search for release substring (e.g., "release 11.5")
132132
import re
133133

134-
m = re.search(r"(?<=release )(\d+)\.(\d+)", nvcc_version_output)
134+
m = re.search(r"release (\d+)\.(\d+)", nvcc_version_output)
135135
if m is not None:
136136
major, minor = map(int, m.groups())
137137
return (major, minor)

0 commit comments

Comments
 (0)