Fix FoldReshapeOpWithUnitExtent generating illegal reshape
authorAhmed Taei <ataei@google.com>
Mon, 19 Apr 2021 23:52:47 +0000 (16:52 -0700)
committerAhmed Taei <ataei@google.com>
Wed, 21 Apr 2021 18:30:45 +0000 (11:30 -0700)
commit10d7924581f8f29c558d089c2546321de26f8849
treee967c4e6972694bb1b97e35c3a82470ec3424894
parent5e6dc5e404f4ddd9667974caf3ba355bd7b01ab2
Fix FoldReshapeOpWithUnitExtent generating illegal reshape

This will prevent fusion that spains all dims and generates
(d0, d1, ...) -> () reshape that isn't legal

Differential Revision: https://reviews.llvm.org/D100805
mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
mlir/test/Dialect/Linalg/drop-unit-extent-dims.mlir