mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
b5b11b2d06
Please review **e2e.py and test_suite belonging to your team**! This is the first part of https://docs.google.com/document/d/16IrwerYi2oJugnRf5hvzukgpJ6FAVEpB6stH_CiNMjY/edit# This PR adds a team name to each test suite. If the name is not specified, it will be reported as unspecified. If you are running a local test, and if the new test suite doesn't have a team name specified, it will raise an exception (in this way, we can avoid missing team names in the future). Note that we will aggregate all of test config into a single file, nightly_test.yaml.
11 lines
No EOL
334 B
YAML
11 lines
No EOL
334 B
YAML
# Test multi-node, multi-GPU Ray SGD example.
|
|
- name: sgd_gpu
|
|
team: ml
|
|
cluster:
|
|
app_config: sgd_gpu/sgd_gpu_app_config.yaml
|
|
compute_template: sgd_gpu/sgd_gpu_compute.yaml
|
|
|
|
run:
|
|
timeout: 3000
|
|
prepare: python wait_cluster.py 2 600
|
|
script: python sgd_gpu/sgd_gpu_test.py --num-workers=2 --use-gpu --address=auto |