[mlir] Prevent segfault in Tensor canonicalization
authorTres Popp <tpopp@google.com>
Wed, 27 Jan 2021 16:28:14 +0000 (17:28 +0100)
committerTres Popp <tpopp@google.com>
Fri, 29 Jan 2021 09:57:58 +0000 (10:57 +0100)
commit0c5e4a25ee232afd0ab21294dfe9ce290957aab6
tree0cb9ed9fe67db31c89b60b22304e544f72256c84
parent1c762a81d20f498a93c38c0f0b10ad7c18650422
[mlir] Prevent segfault in Tensor canonicalization

This segfault could occur from out of bounds accesses when simplifying
tensor.extract with a constant index and a tensor created by
tensor.from_elements.

This IR is not necesarilly invalid as it might conditionally be
never executed.

Differential Revision: https://reviews.llvm.org/D95535
mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
mlir/test/Dialect/Tensor/canonicalize.mlir