[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