forked from dmlc/xgboost
/
pipeline-mgpu.yml
43 lines (43 loc) · 1.36 KB
/
pipeline-mgpu.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
env:
DOCKER_CACHE_ECR_ID: "492475357299"
DOCKER_CACHE_ECR_REGION: "us-west-2"
DISABLE_RELEASE: "1"
# Skip uploading artifacts to S3 bucket
# Also, don't build all CUDA archs; just build sm_75
steps:
- label: ":moneybag: Enforce daily budget"
command: "tests/buildkite/enforce_daily_budget.sh"
key: enforce-daily-budget
agents:
queue: pipeline-loader
- wait
- block: ":rocket: Run this test job"
if: build.pull_request.repository.fork == true
#### -------- CONTAINER BUILD --------
- label: ":docker: Build containers"
commands:
- "tests/buildkite/build-containers.sh gpu"
- "tests/buildkite/build-containers.sh gpu_build_centos7"
- "tests/buildkite/build-containers.sh jvm_gpu_build"
key: build-containers
agents:
queue: linux-amd64-cpu
- wait
#### -------- BUILD --------
- label: ":console: Build CUDA"
command: "tests/buildkite/build-cuda.sh"
key: build-cuda
agents:
queue: linux-amd64-cpu
- label: ":console: Build JVM packages with CUDA"
command: "tests/buildkite/build-jvm-packages-gpu.sh"
key: build-jvm-packages-gpu
agents:
queue: linux-amd64-mgpu
- wait
#### -------- TEST --------
- label: ":console: Test Python package, 4 GPUs"
command: "tests/buildkite/test-python-gpu.sh mgpu"
key: test-python-mgpu
agents:
queue: linux-amd64-mgpu