Build integrated OpenCL Linux wheel in CI

microsoft · May 30, 2022 · c4e6354 · c4e6354
1 parent 6ae3076
commit c4e6354
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.ci/setup.sh b/.ci/setup.sh
@@ -65,7 +65,7 @@ else  # Linux
             libopenmpi-dev \
             openmpi-bin
     fi
-    if [[ $TASK == "gpu" ]]; then
+    if [[ $TASK == "gpu" || $TASK == "bdist" ]]; then
         sudo add-apt-repository ppa:mhier/libboost-latest -y
         sudo apt-get update
         sudo apt-get install --no-install-recommends -y \

diff --git a/.ci/test.sh b/.ci/test.sh
@@ -162,10 +162,11 @@ elif [[ $TASK == "bdist" ]]; then
         else
             PLATFORM="manylinux2014_$ARCH"
         fi
-        cd $BUILD_DIRECTORY/python-package && python setup.py bdist_wheel --plat-name=$PLATFORM --python-tag py3 || exit -1
+        cd $BUILD_DIRECTORY/python-package && python setup.py bdist_wheel --integrated-opencl --plat-name=$PLATFORM --python-tag py3 || exit -1
         if [[ $PRODUCES_ARTIFACTS == "true" ]]; then
             cp dist/lightgbm-$LGB_VER-py3-none-$PLATFORM.whl $BUILD_ARTIFACTSTAGINGDIRECTORY
         fi
+        export LIGHTGBM_TEST_DUAL_CPU_GPU=1
     fi
     pip install --user $BUILD_DIRECTORY/python-package/dist/*.whl || exit -1
     pytest $BUILD_DIRECTORY/tests || exit -1