diff --git a/test/test_ops.py b/test/test_ops.py index 8ea760bc1b..1329818cb8 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -2117,6 +2117,13 @@ class TestOps(unittest.TestCase): lambda x: torch.nn.functional.avg_pool2d(x, kernel_size=(111,28)), lambda x: Tensor.avg_pool2d(x, kernel_size=(111,28)), rtol=1e-5) + # TODO: linearizer block error + @unittest.expectedFailure + def test_avg_pool3d_failure(self): + helper_test_op([(1,1,16,16,16)], + lambda x: torch.nn.functional.avg_pool3d(x, kernel_size=(8,8,8), stride=5, padding=1, count_include_pad=False), + lambda x: Tensor.avg_pool2d(x, kernel_size=(8,8,8), stride=5, padding=1, count_include_pad=False), rtol=1e-5, forward_only=True) + def test_interpolate_linear(self): for in_sz, out_sz in [((52,),(29,)), ((29,),(52,))]: helper_test_op([(2,3)+in_sz],