pipeline.gpu.yml 3.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. # Todo: Enable once tests are available
  2. #- label: ":tv: :octopus: Tune GPU tests "
  3. # conditions: ["RAY_CI_TUNE_AFFECTED"]
  4. # commands:
  5. # - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
  6. # - TUNE_TESTING=1 ./ci/travis/install-dependencies.sh
  7. # - pip install -Ur ./python/requirements_ml_docker.txt
  8. # - ./ci/travis/env_info.sh
  9. # - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-flaky,gpu,gpu_only python/ray/tune/...
  10. - label: ":tv: :brain: RLlib: GPU Examples {A/B}"
  11. conditions: ["RAY_CI_RLLIB_AFFECTED"]
  12. commands:
  13. - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
  14. - RLLIB_TESTING=1 ./ci/travis/install-dependencies.sh
  15. - pip install -Ur ./python/requirements_ml_docker.txt
  16. - ./ci/travis/env_info.sh
  17. # Todo: enable once tests pass
  18. # - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only
  19. # --test_tag_filters=examples_A,examples_B,-flaky --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1 rllib/...
  20. # Todo: enable once tests pass
  21. #- label: ":tv: :brain: RLlib: GPU Examples {C/D}"
  22. # conditions: ["RAY_CI_RLLIB_AFFECTED"]
  23. # commands:
  24. # - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
  25. # - RLLIB_TESTING=1 ./ci/travis/install-dependencies.sh
  26. # - pip install -Ur ./python/requirements_ml_docker.txt
  27. # - ./ci/travis/env_info.sh
  28. # - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only
  29. # --test_tag_filters=examples_C,examples_D,-flaky --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1 rllib/...
  30. # Todo: enable once tests pass
  31. #- label: ":tv: :brain: RLlib: GPU Examples {E/P}"
  32. # conditions: ["RAY_CI_RLLIB_AFFECTED"]
  33. # commands:
  34. # - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
  35. # - RLLIB_TESTING=1 ./ci/travis/install-dependencies.sh
  36. # - pip install -Ur ./python/requirements_ml_docker.txt
  37. # - ./ci/travis/env_info.sh
  38. # - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only
  39. # --test_tag_filters=examples_E,examples_F,examples_G,examples_H,examples_I,examples_J,examples_K,examples_L,examples_M,examples_N,examples_O,examples_P,-flaky --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1
  40. # rllib/...
  41. # Todo: enable once tests pass
  42. #- label: ":tv: :brain: RLlib: GPU Examples {Q/Z}"
  43. # conditions: ["RAY_CI_RLLIB_AFFECTED"]
  44. # commands:
  45. # - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/travis/upload_build_info.sh; fi }; trap cleanup EXIT
  46. # - RLLIB_TESTING=1 ./ci/travis/install-dependencies.sh
  47. # - pip install -Ur ./python/requirements_ml_docker.txt
  48. # - ./ci/travis/env_info.sh
  49. # - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only
  50. # --test_tag_filters=examples_Q,examples_R,examples_S,examples_T,examples_U,examples_V,examples_W,examples_X,examples_Y,examples_Z,-flaky --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1
  51. # rllib/...