#0: Set WH_ARCH_YAML only if we have a wormhole machine

tenstorrent · Sep 16, 2024 · 1f836dc · 1f836dc
1 parent af07d32
commit 1f836dc
Showing 1 changed file with 8 additions and 4 deletions.
diff --git 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