[Mlir-commits] [mlir] [mlir][test] Add TD example for peel+vectorize (depthwise conv) (PR #90200)

Andrzej Warzyński llvmlistbot at llvm.org
Fri Apr 26 09:05:49 PDT 2024


banach-space wrote:

>  wondering if the canonicalization would work in dynamic sizes. 

It should, but I won't have the bandwidth to check. It's not hard to update examples like this, just a bit fiddly, and I need to prioritise sth else in the coming weeks 😅 

> Is it required by decompose ? ( I am not too familiar with the op)

Yes. The vectoriser can only vectorise 1D convs. We deal with 2D convs by decomposing them into 1D convs. That's like a "vectoriser pre-processing" stage.

https://github.com/llvm/llvm-project/pull/90200


More information about the Mlir-commits mailing list