diff options
| author | Jakub Lichman <limo@google.com> | 2020-07-23 12:26:05 +0200 |
|---|---|---|
| committer | Alex Zinenko <zinenko@google.com> | 2020-07-23 12:27:05 +0200 |
| commit | 919922b0c20e99de51bb0d06c98a5fc2fa5feec4 (patch) | |
| tree | 52e8b2681e67590ac2ab104f6e6340649aeef2b1 /llvm/lib/Bitcode/Reader/BitcodeReader.cpp | |
| parent | 722e5d6aaeb5ec0e75506de19e582302c65734cc (diff) | |
[mlir] Added verification check for linalg.conv to ensure memrefs are of rank > 2
linalg.conv does not support memrefs with rank smaller than 3 as stated here:
https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/nn/convolution
However it does not verify it and thus crashes with "LLVM ERROR: out of memory"
error for 1D case and "nWin > 0 && "expected at least one window dimension"" assertion
for 2D case. This commit adds check for that in the verification method.
Differential Revision: https://reviews.llvm.org/D84317
Diffstat (limited to 'llvm/lib/Bitcode/Reader/BitcodeReader.cpp')
0 files changed, 0 insertions, 0 deletions