@@ -56,13 +56,14 @@ def forward(self, x):
5656
5757class TestProfiling :
5858 @pytest .mark .xfail (reason = "SoftMax support PR is not merged so far." , strict = True )
59- def test__softmax (self , caplog ):
59+ def test__softmax (self , caplog , request ):
6060 caplog .set_level (logging .INFO )
6161 model = SoftmaxModule (- 1 )
6262 lower_run_compare (
6363 model ,
6464 (10 ,),
6565 dlg_model_verifier = BaseGraphVerifier (1 , []),
66+ request = request ,
6667 use_profiling = True ,
6768 output_comparator = NumericalStatsOutputComparator (),
6869 )
@@ -77,14 +78,15 @@ def test__softmax(self, caplog):
7778 3 : (), # Neutron Dump
7879 }
7980
80- def test__parallel_pool (self , caplog ):
81+ def test__parallel_pool (self , caplog , request ):
8182 caplog .set_level (logging .INFO )
8283 input_shape = (1 , 3 , 32 , 32 )
8384 model = ParallelPoolModel (input_shape [1 ])
8485 lower_run_compare (
8586 model ,
8687 input_shape ,
8788 dlg_model_verifier = BaseGraphVerifier (1 , []),
89+ request = request ,
8890 output_comparator = NumericalStatsOutputComparator (),
8991 use_neutron_for_format_conversion = False ,
9092 use_profiling = True ,
@@ -105,14 +107,15 @@ def test__parallel_pool(self, caplog):
105107 }
106108
107109 @pytest .mark .xfail (reason = "SoftMax support PR is not merged so far." , strict = True )
108- def test__cifar (self , caplog ):
110+ def test__cifar (self , caplog , request ):
109111 caplog .set_level (logging .INFO )
110112 input_shape = (1 , 3 , 32 , 32 )
111113 model = CifarNetModel ()
112114 lower_run_compare (
113115 model ,
114116 input_shape ,
115117 dlg_model_verifier = BaseGraphVerifier (1 , []),
118+ request = request ,
116119 output_comparator = NumericalStatsOutputComparator (),
117120 use_neutron_for_format_conversion = False ,
118121 use_profiling = True ,
@@ -133,14 +136,15 @@ def test__cifar(self, caplog):
133136 11 : (), # Neutron Dump
134137 }
135138
136- def test__avg_pool (self , caplog ):
139+ def test__avg_pool (self , caplog , request ):
137140 caplog .set_level (logging .INFO )
138141 input_shape = (2 , 9 , 6 , 15 )
139142 model = AvgPool2dModule (False , 0 )
140143 lower_run_compare (
141144 model ,
142145 input_shape ,
143146 dlg_model_verifier = BaseGraphVerifier (1 , []),
147+ request = request ,
144148 output_comparator = NumericalStatsOutputComparator (),
145149 use_neutron_for_format_conversion = False ,
146150 use_profiling = True ,
0 commit comments