Skip to content

Commit d9c1026

Browse files
eukubpull[bot]
authored andcommitted
Improvement of the regular expression
1 parent 2a3dad3 commit d9c1026

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

configure.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1020,7 +1020,7 @@ def get_native_cuda_compute_capabilities(environ_cp):
10201020
if os.path.isfile(device_query_bin) and os.access(device_query_bin, os.X_OK):
10211021
try:
10221022
output = run_shell(device_query_bin).split('\n')
1023-
pattern = re.compile('[0-9]*\\.[0-9]*')
1023+
pattern = re.compile('\d*\\.\d*')
10241024
output = [pattern.search(x) for x in output if 'Capability' in x]
10251025
output = ','.join(x.group() for x in output if x is not None)
10261026
except subprocess.CalledProcessError:

0 commit comments

Comments
 (0)