Merge branch 'mlperf-inference' into mlperf-inference

GATEOverflow · Nov 18, 2024 · 8b42ced · 8b42ced
2 parents a9ae411 + f72ffc1
commit 8b42ced
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/script/get-cudnn/customize.py b/script/get-cudnn/customize.py
@@ -109,7 +109,7 @@ def preprocess(i):
     my_tar.close()
 
     import re
-    version_match = re.match(r'cudnn-.*?-(\d.\d.\d.\d)', folder_name)
+    version_match = re.match(r'cudnn-.*?-(\d+.\d+.\d+.\d+)', folder_name)
     if not version_match:
         return {'return': 1, 'error': 'Extracted CUDNN folder does not seem proper - Version information missing'}
     version = version_match.group(1)

diff --git a/script/get-tensorrt/customize.py b/script/get-tensorrt/customize.py
@@ -100,7 +100,7 @@ def preprocess(i):
     my_tar.close()
 
     import re
-    version_match = re.match(r'TensorRT-(\d.\d.\d.\d)', folder_name)
+    version_match = re.match(r'TensorRT-(\d+.\d+.\d+.\d+)', folder_name)
     if not version_match:
         return {'return': 1, 'error': 'Extracted TensorRT folder does not seem proper - Version information missing'}
     version = version_match.group(1)