aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/CodeGen/ModuloSchedule.cpp
diff options
context:
space:
mode:
authorJakub Lichman <limo@google.com>2020-09-17 09:26:30 +0000
committerJakub Lichman <limo@google.com>2020-09-17 09:39:41 +0000
commit347d59b16c71194d7a9372dd69d3e41ebeca3113 (patch)
tree78cd11522fe977c1bacd7c1db51a7112c99b4037 /llvm/lib/CodeGen/ModuloSchedule.cpp
parent9218f9283802b2d1ff33c490761fdb925b1e56d9 (diff)
downloadllvm-347d59b16c71194d7a9372dd69d3e41ebeca3113.zip
llvm-347d59b16c71194d7a9372dd69d3e41ebeca3113.tar.gz
llvm-347d59b16c71194d7a9372dd69d3e41ebeca3113.tar.bz2
[mlir][Linalg] Convolution tiling added to ConvOp vectorization pass
ConvOp vectorization supports now only convolutions of static shapes with dimensions of size either 3(vectorized) or 1(not) as underlying vectors have to be of static shape as well. In this commit we add support for convolutions of any size as well as dynamic shapes by leveraging existing matmul infrastructure for tiling of both input and kernel to sizes accepted by the previous version of ConvOp vectorization. In the future this pass can be extended to take "tiling mask" as a user input which will enable vectorization of user specified dimensions. Differential Revision: https://reviews.llvm.org/D87676
Diffstat (limited to 'llvm/lib/CodeGen/ModuloSchedule.cpp')
0 files changed, 0 insertions, 0 deletions