[Flow] Add pattern to canonicalize consecutive pads #17878
Conversation
compiler/src/iree/compiler/Dialect/Flow/Transforms/Canonicalizer.cpp
Outdated
Show resolved
Hide resolved
compiler/src/iree/compiler/Dialect/Flow/Transforms/Canonicalizer.cpp
Outdated
Show resolved
Hide resolved
compiler/src/iree/compiler/Dialect/Flow/Transforms/test/flow_canonicalize.mlir
Show resolved
Hide resolved
Currently GlobalOpt and Flow are sharing the same "CanonicalizeExt" pass, so this is the place that makes the most sense (other than upstream, this pattern should probably go upstream). To move it to GlobalOpt would require making a GlobalOpt specific canonicalization pass, and this pattern fits well in both places. |
Would this canonicalization pass be used in the planned dispatch partitioning/preprocessing "phase" #18099? |
|
I would assume so, yes. Flow could potentially just use a default canonicalization pass after that split also. |
compiler/src/iree/compiler/Dialect/Flow/Transforms/test/flow_canonicalize.mlir
Outdated
Show resolved
Hide resolved
This allows fusing `pad(pad)` into a single pad with the same padding value. This is added here due to the inclusion of the Affine dialect and upstream tensor not depending on Affine.
This allows fusing
pad(pad)into a single pad with the same paddingvalue. This is added here due to the inclusion of the Affine dialect and
upstream tensor not depending on Affine.