Add lit test.

Author: sahas3

projects/pt1/python/torch_mlir_e2e_test/test_suite/pooling.py

@@ -1865,10 +1865,9 @@ class AvgPool3dSingleIntTupleParamsModule(torch.nn.Module):
     def __init__(self):
         super().__init__()
         self.apd = torch.nn.AvgPool3d(
-            kernel_size=(6, 6),
+            kernel_size=(6, 6, 6),
             stride=(2,),
-            padding=(1, 1),
-            count_include_pad=False,
+            padding=(1, 1, 1),
         )
 
     @export

test/Dialect/Torch/canonicalize.mlir

@@ -3389,3 +3389,30 @@ func.func @torch.symbolic_int$canonicalize(%arg0: !torch.vtensor<[?],f32>, %arg1
   torch.bind_symbolic_shape %3, [%0], affine_map<()[s0] -> (s0)> : !torch.vtensor<[?],f32>
   return %3 : !torch.vtensor<[?],f32>
 }
+
+// -----
+// CHECK-LABEL:   func.func @torch.aten.avg_pool2d.single_int_tuple(
+// CHECK-SAME:      %[[ARG0:.*]]: !torch.vtensor<[2,4,20,20],f32>) -> !torch.vtensor<[2,4,9,9],f32> {
+// CHECK:           %[[NONE:.*]] = torch.constant.none
+// CHECK:           %[[FALSE:.*]] = torch.constant.bool false
+// CHECK:           %[[C_6:.*]] = torch.constant.int 6
+// CHECK:           %[[C_1:.*]] = torch.constant.int 1
+// CHECK:           %[[C_2:.*]] = torch.constant.int 2
+// CHECK:           %[[KERNEL:.*]] = torch.prim.ListConstruct %[[C_6]], %[[C_6]] : (!torch.int, !torch.int) -> !torch.list<int>
+// CHECK:           %[[STRIDE:.*]] = torch.prim.ListConstruct %[[C_1]], %[[C_1]] : (!torch.int, !torch.int) -> !torch.list<int>
+// CHECK:           %[[PAD:.*]] = torch.prim.ListConstruct %[[C_2]], %[[C_2]] : (!torch.int, !torch.int) -> !torch.list<int>
+// CHECK:           %[[POOL:.*]] = torch.aten.avg_pool2d %[[ARG0]], %[[KERNEL]], %[[PAD]], %[[STRIDE]], %[[FALSE]], %[[FALSE]], %[[NONE]] : !torch.vtensor<[2,4,20,20],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[2,4,9,9],f32>
+// CHECK:           return %[[POOL]]
+func.func @torch.aten.avg_pool2d.single_int_tuple(%arg0: !torch.vtensor<[2,4,20,20],f32>) -> !torch.vtensor<[2,4,9,9],f32> {
+    %int6 = torch.constant.int 6
+    %0 = torch.prim.ListConstruct %int6 : (!torch.int) -> !torch.list<int>
+    %int2 = torch.constant.int 2
+    %1 = torch.prim.ListConstruct %int2 : (!torch.int) -> !torch.list<int>
+    %int1 = torch.constant.int 1
+    %2 = torch.prim.ListConstruct %int1 : (!torch.int) -> !torch.list<int>
+    %false = torch.constant.bool false
+    %false_0 = torch.constant.bool false
+    %none = torch.constant.none
+    %3 = torch.aten.avg_pool2d %arg0, %0, %1, %2, %false, %false_0, %none : !torch.vtensor<[2,4,20,20],f32>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[2,4,9,9],f32>
+    return %3 : !torch.vtensor<[2,4,9,9],f32>
+  }