ray/release/sgd_tests/sgd_tests.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
[Release] Add multi-node, multi-GPU SGD release test (#16046) 2021-05-31 16:23:04 -07:00			`# Test multi-node, multi-GPU Ray SGD example.`
			`- name: sgd_gpu`
[Nightly Test] Add a team column to each test config. (#21198) 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. 2021-12-28 07:42:41 +09:00			`team: ml`
[Release] Add multi-node, multi-GPU SGD release test (#16046) 2021-05-31 16:23:04 -07:00			`cluster:`
[SGD] Add support for native Torch AMP in SGD (#16382) * SGD native AMP initial commit * SGD native amp second pass * Update docs * Update TorchTrainer doc * Temp fix release test * Update release/sgd_tests/sgd_gpu/sgd_gpu_app_config.yaml Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com> 2021-06-16 02:48:21 +02:00			`app_config: sgd_gpu/sgd_gpu_app_config.yaml`
[Release] Add multi-node, multi-GPU SGD release test (#16046) 2021-05-31 16:23:04 -07:00			`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`