Skip to content

Commit 619a718

Browse files
committed
Reduced number of models tested for QNN, and skipped Whisper model
Signed-off-by: Shubham Agrawal <[email protected]>
1 parent ee2faf1 commit 619a718

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

tests/transformers/models/test_causal_lm_models.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,13 @@
4646
"ibm-granite/granite-guardian-3.1-2b",
4747
]
4848

49+
test_models_qnn = [
50+
"mistralai/Mixtral-8x7B-Instruct-v0.1",
51+
"meta-llama/Llama-3.2-1B",
52+
"unsloth/gemma-2b",
53+
"ibm-granite/granite-guardian-3.1-2b",
54+
]
55+
4956
spd_test_models = [
5057
"TinyLlama/TinyLlama-1.1B-Chat-v1.0",
5158
]
@@ -241,7 +248,7 @@ def test_causal_lm_pytorch_vs_kv_vs_ort_vs_ai100(model_name):
241248

242249
@pytest.mark.on_qaic
243250
@pytest.mark.qnn
244-
@pytest.mark.parametrize("model_name", test_models)
251+
@pytest.mark.parametrize("model_name", test_models_qnn)
245252
def test_causal_lm_pytorch_vs_kv_vs_ort_vs_ai100_qnn(model_name):
246253
"""
247254
QNN Compilation Test

tests/transformers/models/test_speech_seq2seq_models.py

+1
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,7 @@ def test_seq2seq_pytorch_vs_kv_vs_ort_vs_ai100(model_name):
366366

367367
@pytest.mark.on_qaic
368368
@pytest.mark.qnn
369+
@pytest.mark.skip(reason="Whisper is currently not supported on QNN")
369370
@pytest.mark.parametrize("model_name", test_models)
370371
def test_seq2seq_pytorch_vs_kv_vs_ort_vs_ai100_qnn(model_name):
371372
"""

0 commit comments

Comments
 (0)