From f8294b3bda76411885ed1985c52565dd67acd8f9 Mon Sep 17 00:00:00 2001
From: geohotstan <135171913+geohotstan@users.noreply.github.com>
Date: Sun, 8 Dec 2024 05:34:38 +0800
Subject: [PATCH] add avg pool 3d failure test (#8105)

* add test

* try simplify test case

* add TODO comment
---
 test/test_ops.py | 7 +++++++
 1 file changed, 7 insertions(+)

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],