Skip to content

Commit 9407d36

Browse files
committed
Update test names and add VGF tests
Change-Id: I25b0b8aca7c2c309a9b19e2308e128f460884a75
1 parent 436e7ac commit 9407d36

File tree

1 file changed

+40
-9
lines changed

1 file changed

+40
-9
lines changed

backends/arm/test/models/test_inception_v3_arm.py

Lines changed: 40 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,11 @@
1111
import torch
1212

1313
from executorch.backends.arm.test.tester.test_pipeline import (
14-
EthosU55PipelineBI,
15-
EthosU85PipelineBI,
16-
TosaPipelineBI,
17-
TosaPipelineMI,
14+
EthosU55PipelineINT,
15+
EthosU85PipelineINT,
16+
TosaPipelineFP,
17+
TosaPipelineINT,
18+
VgfPipeline,
1819
)
1920

2021
from torchvision import models, transforms
@@ -31,8 +32,8 @@
3132

3233

3334
@pytest.mark.slow
34-
def test_ic3_tosa_MI():
35-
pipeline = TosaPipelineMI[input_t](
35+
def test_ic3_tosa_FP():
36+
pipeline = TosaPipelineFP[input_t](
3637
ic3,
3738
model_inputs,
3839
aten_op=[],
@@ -44,7 +45,7 @@ def test_ic3_tosa_MI():
4445

4546
@pytest.mark.slow
4647
def test_ic3_tosa_BI():
47-
pipeline = TosaPipelineBI[input_t](
48+
pipeline = TosaPipelineINT[input_t](
4849
ic3,
4950
model_inputs,
5051
aten_op=[],
@@ -60,7 +61,7 @@ def test_ic3_tosa_BI():
6061
@pytest.mark.skip(reason="Takes too long to run on CI")
6162
@common.XfailIfNoCorstone300
6263
def test_ic3_u55_BI():
63-
pipeline = EthosU55PipelineBI[input_t](
64+
pipeline = EthosU55PipelineINT[input_t](
6465
ic3,
6566
model_inputs,
6667
aten_ops=[],
@@ -77,7 +78,7 @@ def test_ic3_u55_BI():
7778
@pytest.mark.skip(reason="Takes too long to run on CI")
7879
@common.XfailIfNoCorstone320
7980
def test_ic3_u85_BI():
80-
pipeline = EthosU85PipelineBI[input_t](
81+
pipeline = EthosU85PipelineINT[input_t](
8182
ic3,
8283
model_inputs,
8384
aten_ops=[],
@@ -88,3 +89,33 @@ def test_ic3_u85_BI():
8889
qtol=1,
8990
)
9091
pipeline.run()
92+
93+
94+
@pytest.mark.slow
95+
@pytest.mark.skip(reason="Takes too long to run on CI")
96+
@common.SkipIfNoModelConverter
97+
def test_ic3_vgf_FP():
98+
pipeline = VgfPipeline[input_t](
99+
ic3,
100+
model_inputs,
101+
aten_op=[],
102+
exir_op=[],
103+
tosa_version="TOSA-1.0+FP",
104+
use_to_edge_transform_and_lower=True,
105+
)
106+
pipeline.run()
107+
108+
109+
@pytest.mark.slow
110+
@pytest.mark.skip(reason="Takes too long to run on CI")
111+
@common.SkipIfNoModelConverter
112+
def test_ic3_vgf_INT():
113+
pipeline = VgfPipeline[input_t](
114+
ic3,
115+
model_inputs,
116+
aten_op=[],
117+
exir_op=[],
118+
tosa_version="TOSA-1.0+INT",
119+
use_to_edge_transform_and_lower=True,
120+
)
121+
pipeline.run()

0 commit comments

Comments
 (0)