[Frontend] [PaddlePaddle] group_norm adjusts test accuracy (apache#11450

) * group_norm adjusts the check accuracy * remove test skip
dchauhan-arm · May 31, 2022 · efec735 · efec735
1 parent bc14f26
commit efec735
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/python/frontend/paddlepaddle/test_forward.py b/tests/python/frontend/paddlepaddle/test_forward.py
@@ -707,7 +707,6 @@ def forward(self, x, index):
         verify_model(GatherNd(), [x_data, y_data])
 
 
-@pytest.mark.skip(reason="See https://github.com/apache/tvm/issues/11435")
 @tvm.testing.uses_gpu
 def test_forward_group_norm():
     class GroupNorm(nn.Layer):
@@ -722,8 +721,8 @@ def forward(self, inputs):
     for input_shape in input_shapes:
         num_channels = input_shape[1]
         input_data = paddle.uniform(input_shape)
-        verify_model(GroupNorm(num_channels, 1), input_data)
-        verify_model(GroupNorm(num_channels, 2), input_data)
+        verify_model(GroupNorm(num_channels, 1), input_data, rtol=1e-4, atol=1e-4)
+        verify_model(GroupNorm(num_channels, 2), input_data, rtol=1e-4, atol=1e-4)
 
 
 @tvm.testing.uses_gpu