Unverified Commit 82ab0f7f authored by Quinn Dawkins's avatar Quinn Dawkins Committed by GitHub
Browse files

[mlir][linalg] Fix rank-reduced cases for extract/insert slice in DropUnitDims (#74723)

Inferring the reshape reassociation indices for extract/insert slice ops
based on the read sizes of the original slicing op will generate an
invalid expand/collapse shape op for already rank-reduced cases. Instead
just infer from the shape of the slice.

Ported from Differential Revision: https://reviews.llvm.org/D147488
parent c398fa00
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment