From eeb98f0cde1109c7dd1fcad77c0713d2e0ec9c4f Mon Sep 17 00:00:00 2001 From: Colman Glagovich Date: Fri, 8 Nov 2024 07:48:07 -0800 Subject: [PATCH] #14519: Fix test script now that pytest params changed --- models/demos/llama3/demo/simple_vision_demo.py | 2 +- tests/scripts/t3000/run_t3000_demo_tests.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/models/demos/llama3/demo/simple_vision_demo.py b/models/demos/llama3/demo/simple_vision_demo.py index 964554280ee..673c3bc5a73 100644 --- a/models/demos/llama3/demo/simple_vision_demo.py +++ b/models/demos/llama3/demo/simple_vision_demo.py @@ -85,7 +85,7 @@ def create_multimodal_model(mesh_device, max_batch_size, max_seq_len, dtype=ttnn @pytest.mark.parametrize( "enable_trace", (False, True), - ids=["no_trace", "trace"], + ids=["no_trace", "yes_trace"], ) @pytest.mark.parametrize("device_params", [{"trace_region_size": 14951424, "num_command_queues": 2}], indirect=True) def test_llama_multimodal_demo_text( diff --git a/tests/scripts/t3000/run_t3000_demo_tests.sh b/tests/scripts/t3000/run_t3000_demo_tests.sh index 81fe4693094..27d089acd33 100755 --- a/tests/scripts/t3000/run_t3000_demo_tests.sh +++ b/tests/scripts/t3000/run_t3000_demo_tests.sh @@ -90,7 +90,7 @@ run_t3000_llama3_vision_tests() { pip install -r models/demos/llama3/requirements.txt for fake_device in "$n300" "$t3k"; do - FAKE_DEVICE=$fake_device LLAMA_DIR=$llama11b WH_ARCH_YAML=$wh_arch_yaml pytest -n auto models/demos/llama3/demo/multimodal_demo_chat.py -k "tt and 1" --timeout 600; fail+=$? + FAKE_DEVICE=$fake_device LLAMA_DIR=$llama11b WH_ARCH_YAML=$wh_arch_yaml pytest -n auto models/demos/llama3/demo/multimodal_demo_chat.py -k "tt" --timeout 600; fail+=$? echo "LOG_METAL: Llama3 vision tests for $fake_device completed" done