[all-commits] [llvm/llvm-project] c3728d: [mlir] support !shape.value_shape when replace Wit...

[Xiang Li via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c3728d28821e212bd3658261e58e744421668720
      https://github.com/llvm/llvm-project/commit/c3728d28821e212bd3658261e58e744421668720
  Author: Xiang Li <python3kgae at outlook.com>
  Date:   2023-01-23 (Mon, 23 Jan 2023)

  Changed paths:
    M mlir/lib/Dialect/Shape/Transforms/OutlineShapeComputation.cpp
    A mlir/test/Dialect/Shape/arg_with_shape.mlir
    M mlir/test/Dialect/Shape/outline-shape-computation.mlir

  Log Message:
  -----------
  [mlir] support !shape.value_shape when replace WithOp in OutlineShapeComputationPass.

Fixes #60069  https://github.com/llvm/llvm-project/issues/60069

In case like:
  %1 = shape.with_shape %arg1, %0 : !shape.value_shape, !shape.shape
  %2 = shape.value_of %1 : tensor<?xf32>
cannot replace %2 with %arg1.
Transform it into
  %2 = shape.value_of %arg1 : tensor<?xf32>

Differential Revision: https://reviews.llvm.org/D142275