Skip to content

Commit 70d9e94

Browse files
Arm backend: Add vgf to TARGETS in parse_test_names script (#12670)
Signed-off-by: Yufeng Shi <[email protected]>
1 parent fab5687 commit 70d9e94

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

backends/arm/scripts/parse_test_names.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
ALL_EDGE_OPS = SAMPLE_INPUT.keys() | CUSTOM_EDGE_OPS
2727

2828
# Add all targets and TOSA profiles we support here.
29-
TARGETS = ["tosa_MI", "tosa_BI", "u55_BI", "u85_BI"]
29+
TARGETS = ["tosa_MI", "tosa_BI", "u55_BI", "u85_BI", "vgf_INT", "vgf_FP"]
3030

3131

3232
def get_op_name_map():

backends/arm/test/ops/test_add.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def test_add_tensor_u85_BI_2(test_data: input_t2):
189189

190190
@common.parametrize("test_data", Add.test_data)
191191
@common.SkipIfNoModelConverter
192-
def test_add_tensor_vgf_fp(test_data: input_t1):
192+
def test_add_tensor_vgf_FP(test_data: input_t1):
193193
pipeline = VgfPipeline[input_t1](
194194
Add(), test_data(), aten_op, exir_op, tosa_version="TOSA-1.0+FP"
195195
)
@@ -198,7 +198,7 @@ def test_add_tensor_vgf_fp(test_data: input_t1):
198198

199199
@common.parametrize("test_data", Add.test_data)
200200
@common.SkipIfNoModelConverter
201-
def test_add_tensor_vgf_int(test_data: input_t1):
201+
def test_add_tensor_vgf_INT(test_data: input_t1):
202202
pipeline = VgfPipeline[input_t1](
203203
Add(),
204204
test_data(),

0 commit comments

Comments
 (0)