1111import torch
1212
1313from 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
2021from torchvision import models , transforms
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
4647def 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
6263def 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
7980def 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