add trt10.7 to workflow

pytorch · Dec 6, 2024 · c8a2aab · c8a2aab
1 parent 5f1d12f
commit c8a2aab
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/scripts/generate-tensorrt-test-matrix.py b/.github/scripts/generate-tensorrt-test-matrix.py
@@ -41,6 +41,11 @@
             "strip_prefix": "TensorRT-10.6.0.26",
             "sha256": "6c6d92c108a1b3368423e8f69f08d31269830f1e4c9da43b37ba34a176797254",
         },
+        "10.7.0": {
+            "urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.7.0/zip/TensorRT-10.7.0.23.Windows.win10.cuda-12.6.zip",
+            "strip_prefix": "TensorRT-10.7.0.23",
+            "sha256": "fbdef004578e7ccd5ee51fe7f846b57422364a743372fd8f9f1d7dbd33f62879",
+        },
     },
     "linux": {
         "10.4.0": {
@@ -58,6 +63,11 @@
             "strip_prefix": "TensorRT-10.6.0.26",
             "sha256": "33d3c2f3f4c84dc7991a4337a6fde9ed33f5c8e5c4f03ac2eb6b994a382b03a0",
         },
+        "10.7.0": {
+            "urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.7.0/tars/TensorRT-10.7.0.23.Linux.x86_64-gnu.cuda-12.6.tar.gz",
+            "strip_prefix": "TensorRT-10.7.0.23",
+            "sha256": "27d0f7e9af657b9fa19bfe9d62376d4de64182ed6274bde04bf143f56d308ec9",
+        },
     },
 }