From 1377f6a7bc1c47ffea9af3014c19bf398862664a Mon Sep 17 00:00:00 2001
From: Stuti Raizada <sraizada@tenstorrent.com>
Date: Thu, 14 Nov 2024 05:01:57 -0800
Subject: [PATCH] #0: add CI tests

---
 tests/scripts/run_performance.sh                        | 2 ++
 tests/scripts/single_card/run_single_card_demo_tests.sh | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/tests/scripts/run_performance.sh b/tests/scripts/run_performance.sh
index 80db980c993..6c41b5ae336 100755
--- a/tests/scripts/run_performance.sh
+++ b/tests/scripts/run_performance.sh
@@ -48,6 +48,8 @@ run_perf_models_llm_javelin() {
     env pytest -n auto models/demos/falcon7b_common/tests -m $test_marker
     env pytest -n auto models/demos/wormhole/mistral7b/tests -m $test_marker
 
+    env QWEN_DIR=/mnt/MLPerf/tt_dnn-models/qwen/Qwen2-7B-Instruct FAKE_DEVICE=N150 pytest -n auto models/demos/qwen/tests -m $test_marker
+
     # Llama3.1-8B
     llama8b=/mnt/MLPerf/tt_dnn-models/llama/Meta-Llama-3.1-8B-Instruct/
     # Llama3.2-1B
diff --git a/tests/scripts/single_card/run_single_card_demo_tests.sh b/tests/scripts/single_card/run_single_card_demo_tests.sh
index b60c78c6068..af7b8e0538d 100755
--- a/tests/scripts/single_card/run_single_card_demo_tests.sh
+++ b/tests/scripts/single_card/run_single_card_demo_tests.sh
@@ -27,6 +27,9 @@ run_common_func_tests() {
   # Skipping: kills WH cards, check issue #14440
   # WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/wormhole/mistral7b/demo/demo.py --timeout 420; fail+=$?
 
+  # Qwen7B
+  QWEN_DIR=/mnt/MLPerf/tt_dnn-models/qwen/Qwen2-7B-Instruct WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml FAKE_DEVICE=N150 pytest -n auto models/demos/qwen/demo/demo.py --timeout 420; fail+=$?
+
   #VGG11/VGG16
   WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -n auto models/demos/vgg/demo/demo.py --timeout 600; fail+=$?