@@ -179,13 +179,7 @@ def test_native_layer_norm_tosa_INT_a16w8(test_data):
179179 pipeline .run ()
180180
181181
182- @common .parametrize (
183- "test_data" ,
184- test_data_suite ,
185- xfails = {
186- "randn_last_dim" : "MLETORCH-1834 - 16A8W native_layer_norm output diff for certain configurations."
187- },
188- )
182+ @common .parametrize ("test_data" , test_data_suite )
189183@common .XfailIfNoCorstone300
190184def test_native_layer_norm_16a8w_u55_INT (test_data ):
191185 """Test layer_norm with int16 I/O quantization for U55."""
@@ -196,18 +190,13 @@ def test_native_layer_norm_16a8w_u55_INT(test_data):
196190 "torch.ops.aten.sub.Tensor" ,
197191 symmetric_io_quantization = True ,
198192 a16w8_quantization = True ,
199- qtol = 128 ,
193+ qtol = 400 ,
200194 epsilon = 2 ** - 16 ,
201195 )
202196 pipeline .run ()
203197
204198
205- u85_xfails_16a8w = {
206- "randn_last_dim" : "MLETORCH-1834 - 16A8W native_layer_norm output diff for certain configurations." ,
207- }
208-
209-
210- @common .parametrize ("test_data" , test_data_suite , xfails = u85_xfails_16a8w )
199+ @common .parametrize ("test_data" , test_data_suite )
211200@common .XfailIfNoCorstone320
212201def test_native_layer_norm_16a8w_u85_INT (test_data ):
213202 """Test layer_norm with int16 I/O quantization for U85."""
@@ -218,7 +207,7 @@ def test_native_layer_norm_16a8w_u85_INT(test_data):
218207 "torch.ops.aten.sub.Tensor" ,
219208 symmetric_io_quantization = True ,
220209 a16w8_quantization = True ,
221- qtol = 128 ,
210+ qtol = 400 ,
222211 epsilon = 2 ** - 16 ,
223212 )
224213 pipeline .run ()
0 commit comments