Skip to content

Commit b674933

Browse files
Arm backend: Skip multihead_attention VGF tests in upstream
Change-Id: I15b20597bc619ff78f248e3864aedb137a494e15 Signed-off-by: Yufeng Shi <[email protected]>
1 parent 81f8d6f commit b674933

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/arm/test/ops/test_multihead_attention.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ def test_multihead_attention_u85_BI(test_data: input_t1):
112112
"test_data",
113113
test_suite,
114114
)
115+
@common.SkipIfNoModelConverter
115116
def test_multihead_attention_vgf_FP(test_data: input_t1):
116117
test_data_vals, module = test_data()
117118
pipeline = VgfPipeline[input_t1](
@@ -128,6 +129,7 @@ def test_multihead_attention_vgf_FP(test_data: input_t1):
128129
"test_data",
129130
test_suite,
130131
)
132+
@common.SkipIfNoModelConverter
131133
def test_multihead_attention_vgf_INT(test_data: input_t1):
132134
test_data_vals, module = test_data()
133135
pipeline = VgfPipeline[input_t1](

0 commit comments

Comments
 (0)