2021-06-01 17:39:18 +02:00
|
|
|
- name: learning_tests
|
|
|
|
|
cluster:
|
|
|
|
|
app_config: app_config.yaml
|
|
|
|
|
compute_template: 8gpus_64cpus.yaml
|
|
|
|
|
|
|
|
|
|
run:
|
2021-08-03 17:34:27 +01:00
|
|
|
timeout: 14400
|
2021-06-01 17:39:18 +02:00
|
|
|
script: python learning_tests/run_learning_tests.py
|
|
|
|
|
|
2021-08-18 17:21:01 +02:00
|
|
|
- name: multi_gpu_learning_tests
|
2021-06-01 17:39:18 +02:00
|
|
|
cluster:
|
|
|
|
|
app_config: app_config.yaml
|
2021-08-19 22:27:30 +02:00
|
|
|
compute_template: 4gpus_64cpus.yaml
|
2021-06-01 17:39:18 +02:00
|
|
|
|
|
|
|
|
run:
|
2021-08-03 17:34:27 +01:00
|
|
|
timeout: 7200
|
2021-08-18 17:21:01 +02:00
|
|
|
script: python multi_gpu_learning_tests/run_multi_gpu_learning_tests.py
|
|
|
|
|
|
|
|
|
|
# We'll have these as per-PR tests soon.
|
|
|
|
|
# - name: example_scripts_on_gpu_tests
|
|
|
|
|
# cluster:
|
|
|
|
|
# app_config: app_config.yaml
|
|
|
|
|
# compute_template: 1gpu_4cpus.yaml
|
|
|
|
|
|
|
|
|
|
# run:
|
|
|
|
|
# timeout: 7200
|
|
|
|
|
# script: bash unit_gpu_tests/run.sh
|
2021-06-01 17:39:18 +02:00
|
|
|
|
|
|
|
|
- name: stress_tests
|
|
|
|
|
cluster:
|
|
|
|
|
app_config: app_config.yaml
|
|
|
|
|
compute_template: 4gpus_544_cpus.yaml
|
|
|
|
|
|
|
|
|
|
run:
|
2021-08-03 17:34:27 +01:00
|
|
|
timeout: 5400
|
|
|
|
|
script: python stress_tests/run_stress_tests.py
|
