@@ -163,13 +163,12 @@ def test_constant_pad_nd__unsupported_paddings(input_shape, paddings, use_qat):
163163 assert any (node .name == "aten_constant_pad_nd_default" for node in nodes )
164164
165165
166- @pytest .mark .xfail (reason = "EIEX=855" )
167166def test_constant_pad_nd__delegation__formatless__supported_padding (use_qat ):
168167 input_shape = (2 , 4 , 6 , 8 ) # Formatless -> the last dim (8) will be padded.
169168 paddings = [0 , 0 , 1 , 2 , 3 , 4 ] # The last dim is padded using the first 2 paddings.
170169 model = ConstantPadNDModule (paddings )
171170 exec_program = to_quantized_edge_program (
172- model , input_shape , use_qat = use_qat
171+ model , input_shape , use_qat = use_qat , use_new_flow_neutron_c = True
173172 ).exported_program ()
174173
175174 # Make sure the `pad` was delegated.
@@ -192,13 +191,12 @@ def test_constant_pad_nd__delegation__formatless__unsupported_padding(use_qat):
192191 )
193192
194193
195- @pytest .mark .xfail (reason = "Regression in Neutron SW 3.0.1 (AIR-14264)" , strict = True )
196194def test_constant_pad_nd__delegation__channels_first__supported_padding (use_qat ):
197195 input_shape = (2 , 4 , 6 , 8 ) # Channels first -> the second dim (4) will be padded.
198196 paddings = [1 , 2 , 3 , 4 , 0 , 0 ] # The second dim is padded using the paddings[4:6].
199197 model = ConstantPadNDConvModule (paddings )
200198 exec_program = to_quantized_edge_program (
201- model , input_shape , use_qat = use_qat
199+ model , input_shape , use_qat = use_qat , use_new_flow_neutron_c = True
202200 ).exported_program ()
203201
204202 # Make sure the `pad` was delegated.
0 commit comments