From 3540d175cbf84cb3e9443c186dad7b057b55eb4c Mon Sep 17 00:00:00 2001 From: mtairum Date: Fri, 19 Jul 2024 17:45:39 +0000 Subject: [PATCH] #5337: Added missing Mistral prefill tests to CI --- .../mistral7b/tests/test_mistral_attention_prefill.py | 1 + .../wormhole/mistral7b/tests/test_mistral_decoder_prefill.py | 1 + tests/scripts/single_card/run_demos_single_card_wh_common.sh | 4 ++-- 3 files changed, 4 insertions(+), 2 deletions(-) create mode 120000 tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_attention_prefill.py create mode 120000 tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_decoder_prefill.py diff --git a/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_attention_prefill.py b/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_attention_prefill.py new file mode 120000 index 000000000000..f4db1391460e --- /dev/null +++ b/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_attention_prefill.py @@ -0,0 +1 @@ +../../../../../../../../models/demos/wormhole/mistral7b/tests/test_mistral_attention_prefill.py \ No newline at end of file diff --git a/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_decoder_prefill.py b/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_decoder_prefill.py new file mode 120000 index 000000000000..b9de87455d88 --- /dev/null +++ b/tests/nightly/wh_b0_only_eth/models/demos/wormhole/mistral7b/tests/test_mistral_decoder_prefill.py @@ -0,0 +1 @@ +../../../../../../../../models/demos/wormhole/mistral7b/tests/test_mistral_decoder_prefill.py \ No newline at end of file diff --git a/tests/scripts/single_card/run_demos_single_card_wh_common.sh b/tests/scripts/single_card/run_demos_single_card_wh_common.sh index 4612be7fcba3..5be3156a7140 100755 --- a/tests/scripts/single_card/run_demos_single_card_wh_common.sh +++ b/tests/scripts/single_card/run_demos_single_card_wh_common.sh @@ -11,8 +11,8 @@ fi WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto --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_1024_stochastic] # working on both -WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/demo/demo.py::test_demo[instruct_weights] --timeout 420 -WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/demo/demo_with_prefill.py::test_demo[general_weights] --timeout 420 +WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/demo/demo.py --timeout 420 +WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/demo/demo_with_prefill.py --timeout 420 # working on both WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto --disable-warnings -q -s --input-method=cli --cli-input="YOUR PROMPT GOES HERE!" models/demos/wormhole/mamba/demo/demo.py --timeout 420