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

Jacques Pienaar llvmlistbot at llvm.org
Wed Jan 24 17:01: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])
----------------
jpienaar wrote:

Could this be named args?

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


More information about the Mlir-commits mailing list