[ci][fix] Fix cuda_exp ci

.ci/test.sh

@@ -195,40 +195,48 @@ if [[ $TASK == "gpu" ]]; then
     elif [[ $METHOD == "source" ]]; then
         cmake -DUSE_GPU=ON ..
     fi
-elif [[ $TASK == "cuda" || $TASK == "cuda_exp" ]]; then
-    if [[ $TASK == "cuda" ]]; then
+elif [[ $TASK == "cuda" ]]; then
+    if [[ $DEVICE == "cuda" ]]; then
         sed -i'.bak' 's/std::string device_type = "cpu";/std::string device_type = "cuda";/' $BUILD_DIRECTORY/include/LightGBM/config.h
         grep -q 'std::string device_type = "cuda"' $BUILD_DIRECTORY/include/LightGBM/config.h || exit -1  # make sure that changes were really done
-    else
+    elif [[ $DEVICE == "cuda_exp" ]]; then
         sed -i'.bak' 's/std::string device_type = "cpu";/std::string device_type = "cuda_exp";/' $BUILD_DIRECTORY/include/LightGBM/config.h
         grep -q 'std::string device_type = "cuda_exp"' $BUILD_DIRECTORY/include/LightGBM/config.h || exit -1  # make sure that changes were really done
         # by default ``gpu_use_dp=false`` for efficiency. change to ``true`` here for exact results in ci tests
         sed -i'.bak' 's/gpu_use_dp = false;/gpu_use_dp = true;/' $BUILD_DIRECTORY/include/LightGBM/config.h
         grep -q 'gpu_use_dp = true' $BUILD_DIRECTORY/include/LightGBM/config.h || exit -1  # make sure that changes were really done
+    else
+        exit -1
     fi
     if [[ $METHOD == "pip" ]]; then
         cd $BUILD_DIRECTORY/python-package && python setup.py sdist || exit -1
-        if [[ $TASK == "cuda" ]]; then
+        if [[ $DEVICE == "cuda" ]]; then
             pip install --user $BUILD_DIRECTORY/python-package/dist/lightgbm-$LGB_VER.tar.gz -v --install-option=--cuda || exit -1
-        else
+        elif [[ $DEVICE == "cuda_exp" ]]; then
             pip install --user $BUILD_DIRECTORY/python-package/dist/lightgbm-$LGB_VER.tar.gz -v --install-option=--cuda-exp || exit -1
+        else
+            exit -1
         fi
         pytest $BUILD_DIRECTORY/tests/python_package_test || exit -1
         exit 0
     elif [[ $METHOD == "wheel" ]]; then
-        if [[ $TASK == "cuda" ]]; then
+        if [[ $DEVICE == "cuda" ]]; then
             cd $BUILD_DIRECTORY/python-package && python setup.py bdist_wheel --cuda || exit -1
-        else
+        elif [[ $DEVICE == "cuda_exp" ]]; then
             cd $BUILD_DIRECTORY/python-package && python setup.py bdist_wheel --cuda-exp || exit -1
+        else
+            exit -1
         fi
         pip install --user $BUILD_DIRECTORY/python-package/dist/lightgbm-$LGB_VER*.whl -v || exit -1
         pytest $BUILD_DIRECTORY/tests || exit -1
         exit 0
     elif [[ $METHOD == "source" ]]; then
-        if [[ $TASK == "cuda" ]]; then
+        if [[ $DEVICE == "cuda" ]]; then
             cmake -DUSE_CUDA=ON ..
-        else
+        elif [[ $DEVICE == "cuda_exp" ]]; then
             cmake -DUSE_CUDA_EXP=ON ..
+        else
+            exit -1
         fi
     fi
 elif [[ $TASK == "mpi" ]]; then

.github/workflows/cuda.yml

@@ -16,7 +16,7 @@ env:
 
 jobs:
   test:
-    name: ${{ matrix.tree_learner }} ${{ matrix.cuda_version }} ${{ matrix.method }} (linux, ${{ matrix.compiler }}, Python ${{ matrix.python_version }})
+    name: ${{ matrix.device }} ${{ matrix.cuda_version }} ${{ matrix.method }} (linux, ${{ matrix.compiler }}, Python ${{ matrix.python_version }})
     runs-on: [self-hosted, linux]
     timeout-minutes: 60
     strategy:
@@ -27,27 +27,27 @@ jobs:
             compiler: gcc
             python_version: "3.8"
             cuda_version: "11.6.2"
-            tree_learner: cuda
+            device: cuda
           - method: pip
             compiler: clang
             python_version: "3.9"
             cuda_version: "10.0"
-            tree_learner: cuda
+            device: cuda
           - method: wheel
             compiler: gcc
             python_version: "3.10"
             cuda_version: "9.0"
-            tree_learner: cuda
+            device: cuda
           - method: source
             compiler: gcc
             python_version: "3.8"
             cuda_version: "11.6.2"
-            tree_learner: cuda_exp
+            device: cuda_exp
           - method: pip
             compiler: clang
             python_version: "3.9"
             cuda_version: "10.0"
-            tree_learner: cuda_exp
+            device: cuda_exp
     steps:
       - name: Setup or update software on host machine
         run: |
@@ -87,6 +87,7 @@ jobs:
             OS_NAME=${{ env.os_name }}
             COMPILER=${{ matrix.compiler }}
             TASK=${{ env.task }}
+            DEVICE=${{ matrix.device }}
             METHOD=${{ matrix.method }}
             CONDA_ENV=${{ env.conda_env }}
             PYTHON_VERSION=${{ matrix.python_version }}