[PATCH] D134577: [mlir][MemRef] Simplify extract_strided_metadata(allocLikeOp)

[Quentin Colombet via Phabricator via llvm-commits]

qcolombet created this revision.
qcolombet added reviewers: chelini, nicolasvasilache.
qcolombet added a project: MLIR.
Herald added subscribers: bzcheeseman, sdasgup3, Groverkss, wenzhicui, wrengr, cota, teijeong, rdzhabarov, tatianashp, msifontes, jurahul, Kayjukh, grosul1, Joonsoo, liufengdb, aartbik, mgester, arpith-jacob, antiagainst, shauheen, rriddle, mehdi_amini.
Herald added a project: All.
qcolombet requested review of this revision.
Herald added a subscriber: stephenneuendorffer.

Teach the pass that simplifies extract_strided_metadata(other_op(memref))
how to get rid of extract_strided_metadata when they are fed by
allocLikeOp.

For the simplification to happen the allocLikeOp needs to have been
normalized. I.e., no weird offset and strides.

When this is the case, we replace:

  base, offset, sizes, strides =
      extract_strided_metadata(allocLikeOp(allocSizes))

With

  base = reinterpret_cast allocLikeOp(allocSizes) to a flat memref<eltTy>
  offset = 0 
  sizes = allocSizes
  strides#i = prod(allocSizes#j, for j in {i+1..rank-1})

The computation involving dynamic sizes are expanded in affine.apply.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D134577

Files:
  mlir/lib/Dialect/MemRef/Transforms/SimplifyExtractStridedMetadata.cpp
  mlir/test/Dialect/MemRef/simplify-extract-strided-metadata.mlir

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D134577.462628.patch
Type: text/x-patch
Size: 13137 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20220924/2754683c/attachment.bin>