8000 CI Adds arm wheel build in cirrus CI (#25362) · scikit-learn/scikit-learn@5af8c6b · GitHub
[go: up one dir, main page]
Skip to content

Commit 5af8c6b

Browse files
thomasjpfanthomasjpfan
authored
CI Adds arm wheel build in cirrus CI (#25362)
1 parent f45a907 commit 5af8c6b

File tree

4 files changed

+40
-4
lines changed

4 files changed

+40
-4
lines changed

‎.github/workflows/wheels.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,12 +142,12 @@ jobs:
142142
CIBW_REPAIR_WHEEL_COMMAND_WINDOWS: bash build_tools/github/repair_windows_wheels.sh {wheel} {dest_dir}
143143
CIBW_BEFORE_TEST_WINDOWS: bash build_tools/github/build_minimal_windows_image.sh ${{ matrix.python 8000 }}
144144
CIBW_TEST_REQUIRES: pytest pandas threadpoolctl
145-
CIBW_TEST_COMMAND: bash {project}/build_tools/github/test_wheels.sh
145+
CIBW_TEST_COMMAND: bash {project}/build_tools/wheels/test_wheels.sh
146146
CIBW_TEST_COMMAND_WINDOWS: bash {project}/build_tools/github/test_windows_wheels.sh ${{ matrix.python }}
147147
CIBW_BUILD_VERBOSITY: 1
148148
CONDA_HOME: /usr/local/miniconda
149149

150-
run: bash build_tools/github/build_wheels.sh
150+
run: bash build_tools/wheels/build_wheels.sh
151151

152152
- name: Store artifacts
153153
uses: actions/upload-artifact@v3

‎build_tools/cirrus/arm_wheel.yml

Lines changed: 38 additions & 2 deletions
13
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ macos_arm64_wheel_task:
99
SKLEARN_SKIP_NETWORK_TESTS=1
1010
SKLEARN_BUILD_PARALLEL=5
1111
CPU_COUNT=2
12-
CIBW_TEST_COMMAND: bash {project}/build_tools/github/test_wheels.sh
12+
CIBW_TEST_COMMAND: bash {project}/build_tools/wheels/test_wheels.sh
13
CIBW_TEST_REQUIRES: pytest pandas threadpoolctl pytest-xdist
1414
CIBW_BUILD_VERBOSITY: 1
1515
PATH: $HOME/mambaforge/bin/:$PATH
@@ -32,7 +32,43 @@ macos_arm64_wheel_task:
3232
- bash ~/mambaforge.sh -b -p ~/mambaforge
3333

3434
cibuildwheel_script:
35-
- bash build_tools/github/build_wheels.sh
35+
- bash build_tools/wheels/build_wheels.sh
36+
37+
wheels_artifacts:
38+
path: "wheelhouse/*"
39+
40+
linux_arm64_wheel_task:
41+
compute_engine_instance:
42+
image_project: cirrus-images
43+
image: family/docker-builder-arm64
44+
architecture: arm64
45+
platform: linux
46+
cpu: 4
47+
memory: 4G
48+
env:
49+
CONFTEST_PATH: ${CIRRUS_WORKING_DIR}/conftest.py
50+
CONFTEST_NAME: conftest.py
51+
CIBW_ENVIRONMENT: OMP_NUM_THREADS=2
52+
OPENBLAS_NUM_THREADS=2
53+
SKLEARN_SKIP_NETWORK_TESTS=1
54+
SKLEARN_BUILD_PARALLEL=5
55+
CPU_COUNT=4
56+
CIBW_TEST_COMMAND: bash {project}/build_tools/wheels/test_wheels.sh
57+
CIBW_TEST_REQUIRES: pytest pandas threadpoolctl pytest-xdist
58+
CIBW_BUILD_VERBOSITY: 1
59+
matrix:
60+
- env:
61+
CIBW_BUILD: cp38-manylinux_aarch64
62+
- env:
63+
CIBW_BUILD: cp39-manylinux_aarch64
64+
- env:
65+
CIBW_BUILD: cp310-manylinux_aarch64
66+
- env:
67+
CIBW_BUILD: cp311-manylinux_aarch64
68+
69+
cibuildwheel_script:
70+
- apt install -y python3 python-is-python3
71+
- bash build_tools/wheels/build_wheels.sh
3672

3773
wheels_artifacts:
3874
path: "wheelhouse/*"

‎build_tools/github/build_wheels.sh renamed to ‎build_tools/wheels/build_wheels.sh

File renamed without changes.

‎build_tools/github/test_wheels.sh renamed to ‎build_tools/wheels/test_wheels.sh

File renamed without changes.

0 commit comments

Comments
 (0)
0