[Mlir-commits] [mlir] [mlir][tensor] Add new helper hooks for RelayoutOp (PR #109642)

Adam Siemieniuk llvmlistbot at llvm.org
Mon Sep 23 04:05:47 PDT 2024


================
@@ -3987,6 +3987,17 @@ SmallVector<int64_t> PackOp::getStaticTiles() {
   return getStaticTilesImpl(*this);
 }
 
+SmallVector<int64_t> PackOp::getTiledOuterDims() {
+  auto innerDimsPos = getInnerDimsPos();
----------------
adam-smnk wrote:

I wonder if it should first normalize possible permutation of `inner_dims_pos` to be more intuitive.
`getAllOuterDims` is guaranteed to return dims in order w.r.t. the result shape.

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


More information about the Mlir-commits mailing list