[mlir] Loop bounds inference in linalg.generic op improved to support bounds for...
authorJakub Lichman <limo@google.com>
Thu, 23 Jul 2020 08:58:21 +0000 (10:58 +0200)
committerAlex Zinenko <zinenko@google.com>
Thu, 23 Jul 2020 09:01:54 +0000 (11:01 +0200)
commite4dd964df0164651f1804612ad41582fb801607f
tree3f8b568365b7b42d39383f11bd67418e194eb21a
parent59a76d957a2603ee0548293ded170c98e1308c48
[mlir] Loop bounds inference in linalg.generic op improved to support bounds for convolution

Loop bound inference is right now very limited as it supports only permutation maps and thus
it is impossible to implement convolution with linalg.generic as it requires more advanced
loop bound inference. This commits solves it for the convolution case.

Depends On D83158

Differential Revision: https://reviews.llvm.org/D83191
mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
mlir/lib/Dialect/Linalg/Transforms/Loops.cpp
mlir/lib/Dialect/Linalg/Utils/Utils.cpp
mlir/test/Dialect/Linalg/loops.mlir