diff options
| author | Quentin Colombet <quentin.colombet@gmail.com> | 2023-02-23 22:26:15 +0000 |
|---|---|---|
| committer | Thomas Raoux <thomasraoux@google.com> | 2023-02-23 22:27:16 +0000 |
| commit | 17b2e73cb477d42771fbc68a215dde648f3eaaef (patch) | |
| tree | 486fa92d1182ce1cf0593717df6aaae394494063 /llvm/lib/Target/RISCV/Disassembler/RISCVDisassembler.cpp | |
| parent | db288184765c0b4010060ebea1f6de3ac1f66445 (diff) | |
| download | llvm-17b2e73cb477d42771fbc68a215dde648f3eaaef.zip llvm-17b2e73cb477d42771fbc68a215dde648f3eaaef.tar.gz llvm-17b2e73cb477d42771fbc68a215dde648f3eaaef.tar.bz2 | |
[mlir][LinAlg][Transform] Add a transform op for conv2d to im2col
This patch adds patterns to convert `linalg.conv_2d_xxx` operations
into `linalg.generic` (for img2col packing) and `linalg.matmul`.
The meat of the patch comes straight from IREE
(https://github.com/iree-org/iree).
(To the original authors are you okay with that?)
What this patch adds is proper plumbing of the im2col patterns into the
transform dialect.
PS: Feel free to add more reviewers. I wanted to cover the original contributors of im2col in IREE but I'm not sure I got all of them.
Reviewed By: nicolasvasilache, ThomasRaoux
Differential Revision: https://reviews.llvm.org/D144108
Diffstat (limited to 'llvm/lib/Target/RISCV/Disassembler/RISCVDisassembler.cpp')
0 files changed, 0 insertions, 0 deletions