dev

.ci/setup.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 
 
-echo 101
+echo 102
 
 ARCH=$(uname -m)
 
@@ -29,13 +29,13 @@ conda config --set always_yes yes --set changeps1 no
 conda update -q -y conda
 
 
-pip install pytest numpy pandas scipy scikit-learn psutil cloudpickle scikit-build-core ninja cffi pyarrow
+pip install pytest numpy pandas scipy scikit-learn psutil cloudpickle scikit-build-core ninja
 
 
 cd $GITHUB_WORKSPACE/.ci
 git clone --recursive -b docs/install-py https://github.com/microsoft/LightGBM.git
 cd LightGBM
-sh ./build-python.sh install --gpu
+sh ./build-python.sh install --gpu --opencl-include-dir="/usr/local/cuda/include/" --opencl-library="/usr/local/cuda/lib64/libOpenCL.so"
 
 
 pytest $GITHUB_WORKSPACE/tests/python_package_test || exit 1

.github/workflows/cuda.yml

@@ -54,6 +54,11 @@ jobs:
               git \
               sudo \
               curl
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 1
+          submodules: false
       - name: Setup and run tests
         run: |
           export BUILD_DIRECTORY="$GITHUB_WORKSPACE"
@@ -63,8 +68,4 @@ jobs:
           nvidia-smi
 
           # build and test
-          rm -rf $GITHUB_WORKSPACE/LightGBM
-          mkdir -p $GITHUB_WORKSPACE/.ci
-          curl --output $GITHUB_WORKSPACE/.ci/setup.sh "https://raw.githubusercontent.com/microsoft/LightGBM/refs/heads/ci/test/.ci/setup.sh"
-          chmod +x $GITHUB_WORKSPACE/.ci/setup.sh
           $GITHUB_WORKSPACE/.ci/setup.sh || exit 1