[Mlir-commits] [mlir] [mlir][python] enable memref.subview (PR #79393)

Maksim Levental llvmlistbot at llvm.org
Wed Jan 24 18:18:54 PST 2024


================
@@ -88,3 +88,17 @@ def testMemRefAttr():
             memref.global_("objFifo_in0", T.memref(16, T.i32()))
         # CHECK: memref.global @objFifo_in0 : memref<16xi32>
         print(module)
+
+
+# CHECK-LABEL: TEST: testSubViewOpInferReturnTypes
+ at run
+def testSubViewOpInferReturnTypes():
+    with Context() as ctx, Location.unknown(ctx):
+        module = Module.create()
+        with InsertionPoint(module.body):
+            x = memref.alloc(T.memref(10, 10, T.i32()), [], [])
+            # CHECK: %[[ALLOC:.*]] = memref.alloc() : memref<10x10xi32>
+            print(x.owner)
+            y = memref.subview(x, [], [], [1, 1], [3, 3], [1, 1])
----------------
makslevental wrote:

I changed this to abide by `OffsetSizeAndStrideOpInterface` python conventions (i.e. only `offsets`, `sizes`, `strides`).

https://github.com/llvm/llvm-project/pull/79393


More information about the Mlir-commits mailing list