Skip to content

Commit c00919b

Browse files
committed
make all test yamls use multi line commands
1 parent 3285176 commit c00919b

File tree

3 files changed

+26
-8
lines changed

3 files changed

+26
-8
lines changed

tests/pipeline_reorg/galaxy_e2e_tests.yaml

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,18 @@
3333
timeout: 45
3434

3535
- name: Galaxy Fabric Multi-Mesh 4x4 and 2x4s stability tests
36-
cmd: python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && tt-run --rank-binding 4x4_2x4_3_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/perf_microbenchmark/routing/test_tt_fabric --test_config tests/tt_metal/tt_metal/perf_microbenchmark/routing/test_fabric_multi_mesh_stability_short_running.yaml
36+
cmd: |
37+
python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && \
38+
tt-run --rank-binding 4x4_2x4_3_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/perf_microbenchmark/routing/test_tt_fabric --test_config tests/tt_metal/tt_metal/perf_microbenchmark/routing/test_fabric_multi_mesh_stability_short_running.yaml
3739
sku: wh_galaxy
3840
owner_id: U08UBDUKH6Z # Neel Nyamagoudar
3941
team: fabric
4042
timeout: 15
4143

4244
- name: Galaxy Fabric Multi-mesh 2x8 and 2x4s stability tests
43-
cmd: python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && tt-run --rank-binding 2x8_2x4_3_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/perf_microbenchmark/routing/test_tt_fabric --test_config tests/tt_metal/tt_metal/perf_microbenchmark/routing/test_fabric_multi_mesh_stability_short_running.yaml
45+
cmd: |
46+
python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && \
47+
tt-run --rank-binding 2x8_2x4_3_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/perf_microbenchmark/routing/test_tt_fabric --test_config tests/tt_metal/tt_metal/perf_microbenchmark/routing/test_fabric_multi_mesh_stability_short_running.yaml
4448
sku: wh_galaxy
4549
owner_id: U08UBDUKH6Z # Neel Nyamagoudar
4650
team: fabric
@@ -69,7 +73,9 @@
6973
timeout: 20
7074

7175
- name: BH Galaxy Socket Pipeline Latency SendRecv Single Galaxy
72-
cmd: python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && tt-run --rank-binding bh_galaxy_split_4x2_multi_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/multihost/socket_pipeline/multiprocess/unit_tests_pipeline --gtest_filter="*SendRecvPipelineSingleGalaxy*"
76+
cmd: |
77+
python tests/tt_metal/tt_fabric/utils/generate_rank_bindings.py && \
78+
tt-run --rank-binding bh_galaxy_split_4x2_multi_mesh_rank_binding.yaml --mpi-args "--allow-run-as-root --tag-output" ./build/test/tt_metal/multihost/socket_pipeline/multiprocess/unit_tests_pipeline --gtest_filter="*SendRecvPipelineSingleGalaxy*"
7379
sku: bh_galaxy
7480
owner_id: U08UBDUKH6Z # Neel Nyamagoudar
7581
team: fabric

tests/pipeline_reorg/galaxy_perf_tests.yaml

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,9 @@
4747
model-type: Motif
4848

4949
- name: Galaxy DiT Wan2.2 model perf tests
50-
cmd: export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && pytest models/tt_dit/tests/models/wan2_2/test_performance_wan.py -k "wh_4x8sp1tp0 and resolution_720p and t2v" --timeout=1200
50+
cmd: |
51+
export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && \
52+
pytest models/tt_dit/tests/models/wan2_2/test_performance_wan.py -k "wh_4x8sp1tp0 and resolution_720p and t2v" --timeout=1200
5153
model: wan2-2
5254
sku: wh_galaxy
5355
owner_id: U03FJB5TM5Y # Colman Glagovich
@@ -58,7 +60,10 @@
5860
model-type: Wan2.2
5961

6062
- name: Galaxy Mochi model perf tests
61-
cmd: export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && export NO_PROMPT=1 && pytest models/tt_dit/tests/models/mochi/test_performance_mochi.py -k "4x8sp1tp0 " --timeout=1000
63+
cmd: |
64+
export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && \
65+
export NO_PROMPT=1 && \
66+
pytest models/tt_dit/tests/models/mochi/test_performance_mochi.py -k "4x8sp1tp0 " --timeout=1000
6267
model: mochi
6368
sku: wh_galaxy
6469
owner_id: U09ELB03XRU # Stephen Osborne
@@ -69,7 +74,9 @@
6974
model-type: mochi
7075

7176
- name: Galaxy DiT Qwen-Image model perf tests
72-
cmd: export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && pytest models/tt_dit/tests/models/qwenimage/test_performance_qwenimage.py -k "4x8" --timeout=1000
77+
cmd: |
78+
export TT_DIT_CACHE_DIR="/tmp/TT_DIT_CACHE" && \
79+
pytest models/tt_dit/tests/models/qwenimage/test_performance_qwenimage.py -k "4x8" --timeout=1000
7380
model: qwenimage
7481
sku: wh_galaxy
7582
owner_id: U08TED0JM9D # Samuel Adesoye
@@ -80,7 +87,9 @@
8087
model-type: QwenImage
8188

8289
- name: Galaxy Sentence Bert tests
83-
cmd: pytest --timeout 600 models/demos/tg/sentence_bert/tests/device_perf_test.py && python3 models/perf/merge_perf_results.py
90+
cmd: |
91+
pytest --timeout 600 models/demos/tg/sentence_bert/tests/device_perf_test.py && \
92+
python3 models/perf/merge_perf_results.py
8493
model: sentence_bert
8594
sku: wh_galaxy
8695
owner_id: U088413NP0Q # Ashai Reddy

tests/pipeline_reorg/t3k_unit_tests.yaml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,10 @@
4747
model-name: falcon7b
4848

4949
- name: t3k_falcon40b_tests
50-
cmd: ls -lart /mnt/MLPerf/tt_dnn-models/tt/Falcon/tiiuae/falcon-40b-instruct/ && mkdir -p generated/test_reports && run_t3000_falcon40b_tests
50+
cmd: |
51+
ls -lart /mnt/MLPerf/tt_dnn-models/tt/Falcon/tiiuae/falcon-40b-instruct/ && \
52+
mkdir -p generated/test_reports && \
53+
run_t3000_falcon40b_tests
5154
sku: wh_llmbox
5255
owner_id: U053W15B6JF # Djordje Ivanovic
5356
team: models

0 commit comments

Comments
 (0)