Merge pull request #8190 from cupy/admin/add-cuda-import-ci

Add CI to test `cupy.show_config()` pass without CUDA installed

.github/workflows/pretest.yml

@@ -94,15 +94,6 @@ jobs:
       run: |
         python -c 'import cupy, cupyx'
 
-        # Test lazy-imported CUDA Toolkit library.
-        rm cupy/cuda/cufft.*.so
-        rm cupy/random/_generator_api.*.so
-        rm cupy_backends/cuda/libs/cusolver.*.so
-        rm cupy_backends/cuda/libs/cusparse.*.so
-        rm cupy_backends/cuda/libs/curand.*.so
-        rm cupy_backends/cuda/libs/cublas.*.so
-        python -c 'import cupy, cupyx'
-
   build-rocm:
     runs-on: ubuntu-22.04
 

.pfnci/config.pbtxt

@@ -488,6 +488,23 @@ configs {
   }
 }
 
+# CUDA Build & Import without CUDA
+configs {
+  key: "cupy.linux.cuda-build"
+  value {
+    requirement {
+      cpu: 8
+      memory: 50
+      disk: 20
+    }
+    time_limit: {
+      seconds: 3600
+    }
+    environment_variables { key: "GPU" value: "0" }
+    command: ".pfnci/linux/main-flexci.sh cuda-build"
+  }
+}
+
 # BENCHMARK
 configs {
   key: "cupy.linux.benchmark.head"

.pfnci/config.tags.json

@@ -196,6 +196,11 @@
         "full",
         "rapids"
     ],
+    "cupy.linux.cuda-build": [
+        "@push",
+        "full",
+        "mini"
+    ],
     "cupy.linux.array-api": [
         "array-api"
     ],

.pfnci/linux/run.sh

@@ -143,6 +143,8 @@ main() {
       fi
       if [[ "${TARGET}" == *rocm* ]]; then
         docker_args+=(--device=/dev/kfd --device=/dev/dri)
+      elif [[ "${TARGET}" == cuda-build ]]; then
+        docker_args+=()
       else
         docker_args+=(--runtime=nvidia)
       fi

.pfnci/linux/tests/cuda-build.Dockerfile

@@ -0,0 +1 @@
+cuda122.Dockerfile

.pfnci/linux/tests/cuda-build.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+# Based on cuda122.sh
+
+set -uex
+
+ACTIONS="$(dirname $0)/actions"
+. "$ACTIONS/_environment.sh"
+
+export NVCC="ccache nvcc"
+
+export CUPY_ACCELERATORS="cutensor,cub"
+
+# Make the build faster.
+export CUPY_NVCC_GENERATE_CODE="arch=compute_70,code=sm_70"
+
+"$ACTIONS/build.sh"
+
+# Make sure that CuPy can be imported without CUDA Toolkit installed.
+rm -rf /usr/local/cuda*
+pushd /
+python3 -c 'import cupy, cupyx; cupy.show_config(_full=True)'
+popd
+
+"$ACTIONS/cleanup.sh"

.pfnci/matrix.yaml

@@ -565,6 +565,12 @@
   tags: ["@push", "full", "rapids"]
   system: "linux"
 
+- project: "cupy.linux.cuda-build"
+  _extern: true
+  target: "cuda-build"
+  tags: ["@push", "full", "mini"]
+  system: "linux"
+
 - project: "cupy.linux.array-api"
   _extern: true
   target: "array-api"