tree 40fb9dade2057c44f3979cda9e4db0e24463e8aa
parent 87a6325fbe4315f3f24555797f216e96539a9397
author MaheshRavishankar <ravishankarm@google.com> 1620346649 -0700
committer MaheshRavishankar <ravishankarm@google.com> 1620353035 -0700

[mlir][Linalg] Allow folding to rank-zero tensor when using rank-reducing subtensors.

The pattern to convert subtensor ops to their rank-reduced versions
(by dropping unit-dims in the result) can also convert to a zero-rank
tensor. Handle that case.
This also fixes a OOB access bug in the existing pattern for such
cases.

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