[all-commits] [llvm/llvm-project] 0cf844: [MLIR][SCF] Fold dim ops of iter_args to respectiv...

[Prashant Kumar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0cf844759add057f76ca72a611e692eea191c7b7
      https://github.com/llvm/llvm-project/commit/0cf844759add057f76ca72a611e692eea191c7b7
  Author: Prashant Kumar <pk5561 at gmail.com>
  Date:   2024-09-27 (Fri, 27 Sep 2024)

  Changed paths:
    M mlir/lib/Dialect/MemRef/Transforms/ResolveShapedTypeResultDims.cpp
    M mlir/test/Dialect/MemRef/resolve-dim-ops.mlir

  Log Message:
  -----------
  [MLIR][SCF] Fold dim ops of iter_args to respective init_args (#109973)

Fold dim ops of iter_args to dim ops of their respective init args.
E.g.:

```
 %0 = ... : tensor<?x?xf32>
 scf.forall ... shared_outs(%arg0 = %0) -> (tensor<?x?xf32>) {
   %1 = tensor.dim %arg0, %c0 : tensor<?x?xf32>
   ...
 }
```
 is folded to:

```

 %0 = ... : tensor<?x?xf32>
 scf.forall ... shared_outs(%arg0 = %0) -> (tensor<?x?xf32>) {
   %1 = tensor.dim %0, %c0 : tensor<?x?xf32>
   ...
 }
```



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications