Run go generate ./... (#1657)

Signed-off-by: Mattt Zmuda <mattt@replicate.com>
replicate · May 13, 2024 · 2d0fe2a · 2d0fe2a
1 parent 6a281e0
commit 2d0fe2a
Show file tree

Hide file tree

Showing 3 changed files with 151 additions and 13 deletions.
diff --git a/pkg/config/cuda_base_images.json b/pkg/config/cuda_base_images.json
@@ -1,4 +1,32 @@
 [
+  {
+    "Tag": "12.4.1-cudnn-devel-ubuntu22.04",
+    "CUDA": "12.4.1",
+    "CuDNN": "",
+    "IsDevel": true,
+    "Ubuntu": "22.04"
+  },
+  {
+    "Tag": "12.4.1-cudnn-devel-ubuntu20.04",
+    "CUDA": "12.4.1",
+    "CuDNN": "",
+    "IsDevel": true,
+    "Ubuntu": "20.04"
+  },
+  {
+    "Tag": "12.3.2-cudnn9-devel-ubuntu22.04",
+    "CUDA": "12.3.2",
+    "CuDNN": "9",
+    "IsDevel": true,
+    "Ubuntu": "22.04"
+  },
+  {
+    "Tag": "12.3.2-cudnn9-devel-ubuntu20.04",
+    "CUDA": "12.3.2",
+    "CuDNN": "9",
+    "IsDevel": true,
+    "Ubuntu": "20.04"
+  },
   {
     "Tag": "12.2.2-cudnn8-devel-ubuntu22.04",
     "CUDA": "12.2.2",
@@ -139,6 +167,13 @@
     "IsDevel": true,
     "Ubuntu": "18.04"
   },
+  {
+    "Tag": "11.6.1-cudnn8-devel-ubuntu20.04",
+    "CUDA": "11.6.1",
+    "CuDNN": "8",
+    "IsDevel": true,
+    "Ubuntu": "20.04"
+  },
   {
     "Tag": "11.5.2-cudnn8-devel-ubuntu20.04",
     "CUDA": "11.5.2",
@@ -251,4 +286,4 @@
     "IsDevel": true,
     "Ubuntu": "16.04"
   }
-]
+]
diff --git a/pkg/config/tf_compatibility_matrix.json b/pkg/config/tf_compatibility_matrix.json
@@ -1,10 +1,23 @@
 [
+  {
+    "TF": "2.16.1",
+    "TFCPUPackage": "tensorflow==2.16.1",
+    "TFGPUPackage": "tensorflow==2.16.1",
+    "CUDA": "12.3",
+    "CuDNN": "8.9",
+    "Pythons": [
+      "3.9",
+      "3.10",
+      "3.11",
+      "3.12"
+    ]
+  },
   {
     "TF": "2.15.0",
     "TFCPUPackage": "tensorflow==2.15.0",
     "TFGPUPackage": "tensorflow==2.15.0",
     "CUDA": "12.2",
-    "CuDNN": "8.8",
+    "CuDNN": "8.9",
     "Pythons": [
       "3.9",
       "3.10",
@@ -432,4 +445,4 @@
       "3.6"
     ]
   }
-]
+]
diff --git a/pkg/config/torch_compatibility_matrix.json b/pkg/config/torch_compatibility_matrix.json
@@ -1,8 +1,8 @@
 [
   {
-    "Torch": "2.2.1+cpu",
-    "Torchvision": "0.17.1",
-    "Torchaudio": "2.2.1",
+    "Torch": "2.3.0+cpu",
+    "Torchvision": "0.18.0",
+    "Torchaudio": "2.3.0",
     "FindLinks": "",
     "ExtraIndexURL": "https://download.pytorch.org/whl/cpu",
     "CUDA": null,
@@ -15,9 +15,9 @@
     ]
   },
   {
-    "Torch": "2.2.1+cu118",
-    "Torchvision": "0.17.1",
-    "Torchaudio": "2.2.1",
+    "Torch": "2.3.0+cu118",
+    "Torchvision": "0.18.0",
+    "Torchaudio": "2.3.0",
     "FindLinks": "",
     "ExtraIndexURL": "https://download.pytorch.org/whl/cu118",
     "CUDA": "11.8",
@@ -30,9 +30,9 @@
     ]
   },
   {
-    "Torch": "2.2.1+cu121",
-    "Torchvision": "0.17.1",
-    "Torchaudio": "2.2.1",
+    "Torch": "2.3.0+cu121",
+    "Torchvision": "0.18.0",
+    "Torchaudio": "2.3.0",
     "FindLinks": "",
     "ExtraIndexURL": "https://download.pytorch.org/whl/cu121",
     "CUDA": "12.1",
@@ -44,6 +44,96 @@
       "3.9"
     ]
   },
+  {
+    "Torch": "2.2.2",
+    "Torchvision": "0.17.2",
+    "Torchaudio": "2.2.2",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu118",
+    "CUDA": "11.8",
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
+  {
+    "Torch": "2.2.2",
+    "Torchvision": "0.17.2",
+    "Torchaudio": "2.2.2",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu121",
+    "CUDA": "12.1",
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
+  {
+    "Torch": "2.2.2",
+    "Torchvision": "0.17.2",
+    "Torchaudio": "2.2.2",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cpu",
+    "CUDA": null,
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
+  {
+    "Torch": "2.2.1",
+    "Torchvision": "0.17.1",
+    "Torchaudio": "2.2.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu118",
+    "CUDA": "11.8",
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
+  {
+    "Torch": "2.2.1",
+    "Torchvision": "0.17.1",
+    "Torchaudio": "2.2.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu121",
+    "CUDA": "12.1",
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
+  {
+    "Torch": "2.2.1",
+    "Torchvision": "0.17.1",
+    "Torchaudio": "2.2.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cpu",
+    "CUDA": null,
+    "Pythons": [
+      "3.7",
+      "3.8",
+      "3.9",
+      "3.10",
+      "3.11"
+    ]
+  },
   {
     "Torch": "2.2.0",
     "Torchvision": "0.17.0",
@@ -1274,4 +1364,4 @@
       "3.11"
     ]
   }
-]
+]