Merge pull request #6794 from kmaehashi/bp-6767-v10-cuda-11.7

[backport] Support CUDA 11.7
cupy · Jun 21, 2022 · 4cacb71 · 4cacb71
2 parents 794f635 + e5492b8
commit 4cacb71
Show file tree

Hide file tree

Showing 16 changed files with 803 additions and 388 deletions.
diff --git a/.pfnci/config.pbtxt b/.pfnci/config.pbtxt
@@ -307,6 +307,42 @@ configs {
   }
 }
 
+# CUDA 11.7 | Linux
+configs {
+  key: "cupy.linux.cuda117"
+  value {
+    requirement {
+      cpu: 8
+      memory: 50
+      disk: 10
+      gpu: 1
+    }
+    time_limit: {
+      seconds: 21600
+    }
+    environment_variables { key: "GPU" value: "1" }
+    command: ".pfnci/linux/main-flexci.sh cuda117"
+  }
+}
+
+# CUDA 11.7 (Multi-GPU) | Linux
+configs {
+  key: "cupy.linux.cuda117.multi"
+  value {
+    requirement {
+      cpu: 8
+      memory: 50
+      disk: 10
+      gpu: 2
+    }
+    time_limit: {
+      seconds: 21600
+    }
+    environment_variables { key: "GPU" value: "2" }
+    command: ".pfnci/linux/main-flexci.sh cuda117.multi"
+  }
+}
+
 # CUDA 11.x + CUDA Python
 configs {
   key: "cupy.linux.cuda11x-cuda-python"

diff --git a/.pfnci/config.tags.json b/.pfnci/config.tags.json
@@ -68,6 +68,17 @@
     ],
     "cupy.linux.cuda116": [],
     "cupy.linux.cuda116.multi": [],
+    "cupy.linux.cuda117": [
+        "@push",
+        "full",
+        "mini"
+    ],
+    "cupy.linux.cuda117.multi": [
+        "@push",
+        "full",
+        "mini",
+        "multi"
+    ],
     "cupy.linux.rocm-4-0": [],
     "cupy.linux.cuda-slow": [
         "slow"