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