HomePhabricator

[mlir] Conv ops vectorization pass

Authored by limo1996 on Sep 8 2020, 1:31 AM.

Description

[mlir] Conv ops vectorization pass

In this commit a new way of convolution ops lowering is introduced.
The conv op vectorization pass lowers linalg convolution ops
into vector contractions. This lowering is possible when conv op
is first tiled by 1 along specific dimensions which transforms
it into dot product between input and kernel subview memory buffers.
This pass converts such conv op into vector contraction and does
all necessary vector transfers that make it work.

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

Details

Committed
limo1996Sep 8 2020, 1:47 AM
Differential Revision
D86619: [mlir] Conv ops vectorization pass
Parents
rG38778e1087b2: Provide anchor for compiler extensions
Branches
Unknown
Tags
Unknown