[all-commits] [llvm/llvm-project] 42c195: [mlir][Shape] Allow shape.split_at to return exten...

[Benjamin Kramer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 42c195f0ec8f2d9236b237c5ad2c6f3ca9b4184c
      https://github.com/llvm/llvm-project/commit/42c195f0ec8f2d9236b237c5ad2c6f3ca9b4184c
  Author: Benjamin Kramer <benny.kra at googlemail.com>
  Date:   2021-03-08 (Mon, 08 Mar 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
    M mlir/lib/Conversion/ShapeToStandard/ShapeToStandard.cpp
    M mlir/test/Conversion/ShapeToStandard/shape-to-standard.mlir

  Log Message:
  -----------
  [mlir][Shape] Allow shape.split_at to return extent tensors and lower it to std.subtensor

split_at can return an error if the split index is out of bounds. If the
user knows that the index can never be out of bounds it's safe to use
extent tensors. This has a straight-forward lowering to std.subtensor.

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