Skip to content

Commit c0dd42f

Browse files
committed
Feat: Support ARM64 platform in XGBoost examples
Signed-off-by: tico88612 <[email protected]>
1 parent e31d11f commit c0dd42f

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

.github/workflows/publish-example-images.yaml

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,7 @@ jobs:
1010
uses: ./.github/workflows/build-and-publish-images.yaml
1111
with:
1212
component-name: ${{ matrix.component-name }}
13-
# TODO (tenzen-y): Support linux/arm64 platform
14-
platforms: linux/amd64
13+
platforms: ${{ matrix.platforms }}
1514
dockerfile: ${{ matrix.dockerfile }}
1615
context: ${{ matrix.context }}
1716
secrets:
@@ -42,12 +41,27 @@ jobs:
4241
dockerfile: examples/pytorch/cpu-demo/Dockerfile
4342
context: examples/pytorch/cpu-demo
4443
- component-name: xgboost-dist-iris
44+
platforms: linux/amd64
4545
dockerfile: examples/xgboost/xgboost-dist/Dockerfile
4646
context: examples/xgboost/xgboost-dist
4747
- component-name: lightgbm-dist-py-test
48+
platforms: linux/amd64
4849
dockerfile: examples/xgboost/lightgbm-dist/Dockerfile
4950
context: examples/xgboost/lightgbm-dist
5051
- component-name: xgboost-dist-rabit-test
52+
platforms: linux/amd64
53+
dockerfile: examples/xgboost/smoke-dist/Dockerfile
54+
context: examples/xgboost/smoke-dist
55+
- component-name: xgboost-dist-iris
56+
platforms: linux/arm64
57+
dockerfile: examples/xgboost/xgboost-dist/Dockerfile
58+
context: examples/xgboost/xgboost-dist
59+
- component-name: lightgbm-dist-py-test
60+
platforms: linux/arm64
61+
dockerfile: examples/xgboost/lightgbm-dist/Dockerfile
62+
context: examples/xgboost/lightgbm-dist
63+
- component-name: xgboost-dist-rabit-test
64+
platforms: linux/arm64
5165
dockerfile: examples/xgboost/smoke-dist/Dockerfile
5266
context: examples/xgboost/smoke-dist
5367
- component-name: mxnet-gpu

0 commit comments

Comments
 (0)