mirror of
https://github.com/vale981/ray
synced 2025-03-10 21:36:39 -04:00
3dc176c42e
* [ci/tune] Add Tune GPU pipeline step to CI * cont. * add sgd gpu tests * format yaml, fix imports * install horovod; fix line wrapping * set GPU per worker to 0.5 * fix import * move test to 4gpu machine * fix lint * lint * set visible devices * pull in tf gpu fix * Fix Tune GPU pipeline step * nit * Disable GPU tests until we have some * Re-add empty rllib tests Co-authored-by: Matthew Deng <matthew.j.deng@gmail.com>
8 lines
518 B
YAML
8 lines
518 B
YAML
- label: ":tv: :octopus: SGD GPU tests "
|
|
conditions: ["RAY_CI_SGD_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/util/sgd/...
|