#5337: Update Mistral-7B CI tests

tenstorrent · Jun 5, 2024 · e816db1 · e816db1
1 parent 5094e68
commit e816db1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/models/demos/wormhole/mistral7b/README.md b/models/demos/wormhole/mistral7b/README.md
@@ -54,12 +54,12 @@ Mistral-7B is running on a single chip. If you are running on a T3000 please set
 
 ```
 # Run the demo with a pre-written batch of 32 user prompts:
-pytest --collect-only models/demos/wormhole/mistral7b/demo/demo.py::test_demo[general_weights]
+pytest models/demos/wormhole/mistral7b/demo/demo.py::test_demo[general_weights]
 ```
 
 We also provide an input file with 32 user question-prompt for instruct weights (don't forget to update your env flags to the correct instruct weights folder):
 ```
-pytest --collect-only models/demos/wormhole/mistral7b/demo/demo.py::test_demo[instruct_weights]
+pytest models/demos/wormhole/mistral7b/demo/demo.py::test_demo[instruct_weights]
 ```
 
 Both input files are provided inside `models/demos/wormhole/mistral7b/demo/`.

diff --git a/tests/scripts/run_performance.sh b/tests/scripts/run_performance.sh
@@ -39,7 +39,7 @@ run_perf_models_llm_javelin() {
         env pytest models/demos/mamba/tests -m $test_marker
     fi
 
-    env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/wormhole/mistral7b/tests -m $test_marker  # -> hanging: issue #7540
+    env pytest models/demos/wormhole/mistral7b/tests -m $test_marker
 
     ## Merge all the generated reports
     env python models/perf/merge_perf_results.py

diff --git a/tests/scripts/single_card/nightly/run_wh_b0_only.sh b/tests/scripts/single_card/nightly/run_wh_b0_only.sh
@@ -25,6 +25,6 @@ env pytest models/demos/mamba/tests/test_mamba_demo.py
 env pytest models/demos/wormhole/mistral7b/tests/test_mistral_embedding.py
 env pytest models/demos/wormhole/mistral7b/tests/test_mistral_rms_norm.py
 env pytest models/demos/wormhole/mistral7b/tests/test_mistral_mlp.py
-env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/wormhole/mistral7b/tests/test_mistral_attention.py
-env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/demos/wormhole/mistral7b/tests/test_mistral_decoder.py
+env pytest models/demos/wormhole/mistral7b/tests/test_mistral_attention.py
+env pytest models/demos/wormhole/mistral7b/tests/test_mistral_decoder.py
 #env WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest models/experimental/functional_unet/tests
diff --git a/tests/scripts/single_card/run_demos_single_card_n150_tests.sh b/tests/scripts/single_card/run_demos_single_card_n150_tests.sh
@@ -13,7 +13,7 @@ export WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml
 pytest --disable-warnings -q -s --input-method=cli --cli-input="YOUR PROMPT GOES HERE!"  models/demos/wormhole/falcon7b/demo_wormhole.py::test_demo[user_input0-default_mode_stochastic]
 
 # working on both
-pytest --disable-warnings -q -s --input-method=cli --cli-input="YOUR PROMPT GOES HERE!"  models/demos/wormhole/mistral7b/demo/demo.py
+pytest models/demos/wormhole/mistral7b/demo/demo.py::test_demo[instruct_weights]
 
 # working on both
 pytest --disable-warnings -q -s --input-method=cli --cli-input="YOUR PROMPT GOES HERE!"  models/demos/mamba/demo/demo.py