30 lines
873 B
YAML
30 lines
873 B
YAML
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:
|
|
- block: ":rocket: Run this test job"
|
|
if: build.pull_request.repository.fork == true
|
|
#### -------- 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
|