mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
21 lines
1.3 KiB
YAML
21 lines
1.3 KiB
YAML
- label: ":tv: :steam_locomotive: Train GPU tests "
|
|
conditions: ["RAY_CI_TRAIN_AFFECTED"]
|
|
commands:
|
|
- cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
|
|
- SGD_TESTING=1 INSTALL_HOROVOD=1 ./ci/travis/install-dependencies.sh
|
|
- pip install -Ur ./python/requirements_ml_docker.txt
|
|
- ./ci/travis/env_info.sh
|
|
- bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=gpu,gpu_only python/ray/train/...
|
|
|
|
- label: ":tv: :brain: RLlib: Multi-GPU Tests"
|
|
conditions: ["RAY_CI_RLLIB_AFFECTED"]
|
|
commands:
|
|
- cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
|
|
- RLLIB_TESTING=1 ./ci/travis/install-dependencies.sh
|
|
- pip install -Ur ./python/requirements_ml_docker.txt
|
|
- ./ci/travis/env_info.sh
|
|
# --jobs 2 is necessary as we only need to have at least 2 gpus on the machine
|
|
# and running tests in parallel would cause timeouts as the other scripts would
|
|
# wait for the GPU to become available.
|
|
- bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --jobs 2
|
|
--test_tag_filters=multi_gpu,-flaky --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 rllib/...
|