[all-commits] [llvm/llvm-project] 1b00b9: [mlir][tosa] Tosa shape propagation for tosa.cond_if

[rsuderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 1b00b94ffc2d60e07ec8e486dad0fcbcbfb99c62
      https://github.com/llvm/llvm-project/commit/1b00b94ffc2d60e07ec8e486dad0fcbcbfb99c62
  Author: Rob Suderman <rob.suderman at gmail.com>
  Date:   2021-08-03 (Tue, 03 Aug 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
    A mlir/include/mlir/Dialect/Tosa/Utils/ShapeUtils.h
    M mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
    M mlir/lib/Dialect/Tosa/Transforms/TosaInferShapes.cpp
    M mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir

  Log Message:
  -----------
  [mlir][tosa] Tosa shape propagation for tosa.cond_if

We can propagate the shape from tosa.cond_if operands into the true/false
regions then through the connected blocks. Then, using the tosa.yield ops
we can determine what all possible return types are.

Reviewed By: jpienaar

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