[all-commits] [llvm/llvm-project] d876e3: [MLIR][Shape] Lower `shape.get_extent` to `extract...

[frgossen via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: d876e3202af3057cc180d7540d0de8b20234f114
      https://github.com/llvm/llvm-project/commit/d876e3202af3057cc180d7540d0de8b20234f114
  Author: Frederik Gossen <frgossen at google.com>
  Date:   2020-06-29 (Mon, 29 Jun 2020)

  Changed paths:
    M mlir/lib/Conversion/ShapeToStandard/ShapeToStandardPatterns.td
    M mlir/test/Conversion/ShapeToStandard/shape-to-standard.mlir

  Log Message:
  -----------
  [MLIR][Shape] Lower `shape.get_extent` to `extract_element` when possible

When the origin of a shape is an extent tensor the operation `get_extent` can be
lowered directly to `extract_element`.
This choice circumvents the necessity to materialize the shape in memory.

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