From 1f836dce0cea0d3e17980ff746bbe564c5f080a4 Mon Sep 17 00:00:00 2001 From: Raymond Kim Date: Mon, 16 Sep 2024 13:40:23 +0000 Subject: [PATCH] #0: Set WH_ARCH_YAML only if we have a wormhole machine --- tests/scripts/run_performance.sh | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/scripts/run_performance.sh b/tests/scripts/run_performance.sh index 467e96ecbc1..0527bc69e6b 100755 --- a/tests/scripts/run_performance.sh +++ b/tests/scripts/run_performance.sh @@ -35,14 +35,18 @@ run_perf_models_llm_javelin() { local tt_arch=$1 local test_marker=$2 - env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/falcon7b_common/tests -m $test_marker + if [ "$tt_arch" == "wormhole_b0" ]; then + export WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml + fi + + env pytest -n auto models/demos/falcon7b_common/tests -m $test_marker if [ "$tt_arch" == "wormhole_b0" ]; then - env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mamba/tests -m $test_marker + env pytest -n auto models/demos/wormhole/mamba/tests -m $test_marker fi - env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/tests -m $test_marker - env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/llama31_8b/tests -m $test_marker + env pytest -n auto models/demos/wormhole/mistral7b/tests -m $test_marker + env pytest -n auto models/demos/wormhole/llama31_8b/tests -m $test_marker ## Merge all the generated reports env python models/perf/merge_perf_results.py