mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
115 lines
No EOL
2.6 KiB
YAML
115 lines
No EOL
2.6 KiB
YAML
- name: horovod_user_test_latest
|
|
cluster:
|
|
app_config: horovod/app_config.yaml
|
|
compute_template: horovod/compute_tpl.yaml
|
|
|
|
|
|
driver_setup: horovod/driver_setup_latest.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python horovod/horovod_user_test.py
|
|
|
|
- name: horovod_user_test_master
|
|
cluster:
|
|
app_config: ../horovod_tests/app_config_master.yaml
|
|
compute_template: horovod/compute_tpl.yaml
|
|
|
|
driver_setup: horovod/driver_setup_master.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python horovod/horovod_user_test.py
|
|
|
|
|
|
- name: train_tensorflow_mnist_test
|
|
cluster:
|
|
app_config: train/app_config.yaml
|
|
compute_template: train/compute_tpl.yaml
|
|
|
|
driver_setup: train/driver_setup.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
timeout: 36000
|
|
script: python train/train_tensorflow_mnist_test.py
|
|
|
|
- name: train_torch_linear_test
|
|
cluster:
|
|
app_config: train/app_config.yaml
|
|
compute_template: train/compute_tpl.yaml
|
|
|
|
driver_setup: train/driver_setup.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
timeout: 36000
|
|
script: python train/train_torch_linear_test.py
|
|
|
|
|
|
- name: xgboost_gpu_connect_latest
|
|
cluster:
|
|
app_config: xgboost/app_config_gpu.yaml
|
|
compute_template: xgboost/tpl_gpu_small_scaling.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
timeout: 1200
|
|
script: python xgboost/train_gpu_connect.py
|
|
|
|
- name: xgboost_gpu_connect_master
|
|
cluster:
|
|
app_config: xgboost/app_config_gpu_master.yaml
|
|
compute_template: xgboost/tpl_gpu_small_scaling.yaml
|
|
|
|
run:
|
|
use_connect: True
|
|
timeout: 1200
|
|
script: python xgboost/train_gpu_connect.py
|
|
|
|
- name: ray_lightning_user_test_latest
|
|
cluster:
|
|
app_config: ray-lightning/app_config.yaml
|
|
compute_template: ray-lightning/compute_tpl.yaml
|
|
|
|
driver_setup: ray-lightning/driver_setup.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python ray-lightning/ray_lightning_user_test.py
|
|
|
|
|
|
- name: ray_lightning_user_test_master
|
|
cluster:
|
|
app_config: ray-lightning/app_config_master.yaml
|
|
compute_template: ray-lightning/compute_tpl.yaml
|
|
|
|
|
|
driver_setup: ray-lightning/driver_setup.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python ray-lightning/ray_lightning_user_test.py
|
|
|
|
|
|
- name: tune_rllib_connect_test
|
|
cluster:
|
|
app_config: ../rllib_tests/app_config.yaml
|
|
compute_template: tune_rllib/compute_tpl.yaml
|
|
|
|
|
|
driver_setup: tune_rllib/driver_setup.sh
|
|
|
|
run:
|
|
use_connect: True
|
|
autosuspend_mins: 10
|
|
timeout: 1200
|
|
script: python tune_rllib/run_connect_tests.py |