[all-commits] [llvm/llvm-project] 7c984b: [mlir] Propagate arith.index_cast past tensor.extract

[Rob Suderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 7c984be21a350d2fd227fd95f36a70165a523b99
      https://github.com/llvm/llvm-project/commit/7c984be21a350d2fd227fd95f36a70165a523b99
  Author: Rob Suderman <rob.suderman at gmail.com>
  Date:   2022-01-25 (Tue, 25 Jan 2022)

  Changed paths:
    M mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
    M mlir/test/Dialect/Tensor/canonicalize.mlir

  Log Message:
  -----------
  [mlir] Propagate arith.index_cast past tensor.extract

If we are extracting it is more useful to push the index_cast past the
extraction. This increases the chance the tensor.extract can evaluated at
compile time.

Reviewed By: rriddle

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