diff --git a/tests/python/frontend/paddlepaddle/test_forward.py b/tests/python/frontend/paddlepaddle/test_forward.py index e381ab5a2f60..56ec3a4e5469 100644 --- 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