add dilation topi tests FINALLY

apache · Apr 26, 2021 · 0881ced · 0881ced
1 parent 60f25d5
commit 0881ced
Showing 1 changed file with 120 additions and 0 deletions.
diff --git a/tests/python/topi/python/test_topi_pooling.py b/tests/python/topi/python/test_topi_pooling.py
@@ -480,6 +480,46 @@ def test_pool3d():
         True,
     )
 
+    # Test non-1 dilation
+    verify_pool3d(
+        [1, 16, 31, 31, 31],
+        [3, 3, 3],
+        [3, 3, 3],
+        [3, 3, 3],
+        [2, 1, 0, 5, 4, 3],
+        "avg",
+        False,
+        True,
+    )
+    verify_pool3d(
+        [1, 16, 32, 32, 32],
+        [2, 2, 2],
+        [2, 2, 2],
+        [2, 2, 2],
+        [0, 5, 4, 3, 2, 1],
+        "avg",
+        False,
+        False,
+    )
+    verify_pool3d(
+        [1, 16, 31, 31, 31],
+        [3, 3, 3],
+        [3, 3, 3],
+        [2, 1, 3],
+        [1, 0, 5, 4, 3, 2],
+        "max",
+        False,
+    )
+    verify_pool3d(
+        [1, 16, 31, 31, 31],
+        [3, 3, 3],
+        [3, 3, 3],
+        [2, 2, 3],
+        [3, 2, 1, 0, 5, 4],
+        "max",
+        True,
+    )
+
 
 def verify_pool2d(
     input_shape, kernel, stride, dilation, padding, pool_type, ceil_mode, count_include_pad=True
@@ -618,6 +658,46 @@ def test_pool2d():
         True,
     )
 
+    # Test non-1 dilations
+    verify_pool2d(
+        [1, 16, 31, 31],
+        [3, 3],
+        [3, 3],
+        [2, 1],
+        [2, 1, 0, 3],
+        "avg",
+        False,
+        True,
+    )
+    verify_pool2d(
+        [1, 16, 32, 32],
+        [2, 3],
+        [2, 2],
+        [2, 3],
+        [0, 3, 2, 1],
+        "avg",
+        False,
+        False,
+    )
+    verify_pool2d(
+        [1, 16, 31, 31],
+        [3, 3],
+        [3, 3],
+        [3, 3],
+        [1, 0, 3, 2],
+        "max",
+        False,
+    )
+    verify_pool2d(
+        [1, 16, 31, 31],
+        [3, 3],
+        [3, 3],
+        [2, 2],
+        [3, 2, 1, 0],
+        "max",
+        True,
+    )
+
 
 def verify_pool1d(
     input_shape,
@@ -764,6 +844,46 @@ def test_pool1d():
         True,
     )
 
+    # Test non-1 dilations
+    verify_pool1d(
+        [1, 16, 31],
+        [3],
+        [3],
+        [2],
+        [2, 5],
+        "avg",
+        False,
+        True,
+    )
+    verify_pool1d(
+        [1, 16, 32],
+        [2],
+        [2],
+        [3],
+        [0, 3],
+        "avg",
+        False,
+        False,
+    )
+    verify_pool1d(
+        [1, 16, 31],
+        [3],
+        [3],
+        [2],
+        [1, 4],
+        "max",
+        False,
+    )
+    verify_pool1d(
+        [1, 16, 31],
+        [3],
+        [3],
+        [3],
+        [3, 0],
+        "max",
+        True,
+    )
+
 
 if __name__ == "__main__":
     test_pool1d()