diff options
| author | Nicolas Vasilache <ntv@google.com> | 2019-12-09 09:14:05 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-09 09:14:39 -0800 |
| commit | 91c0074624a2ba7f3f56617da75afe70713b82b2 (patch) | |
| tree | e28afd4a368fbefb3ccea8fc5b9f3197f3c967a0 /mlir/lib/Transforms/LoopTiling.cpp | |
| parent | 70aeb4566e35541ffeef28050babc1c9580b43eb (diff) | |
| download | bcm5719-llvm-91c0074624a2ba7f3f56617da75afe70713b82b2.tar.gz bcm5719-llvm-91c0074624a2ba7f3f56617da75afe70713b82b2.zip | |
[StructuredOps][Linalg] Add a primitive pattern to rewrite the linalg.generic form of matmul to vector form.
This CL uses the newly expanded matcher support to easily detect when a linalg.generic has a multiply-accumulate body. A linalg.generic with such a body is rewritten as a vector contraction.
This CL additionally limits the rewrite to the case of matrix multiplication on contiguous and statically shaped memrefs for now.
Before expanding further, we should harden the infrastructure for expressing custom ops with the structured ops abstraction.
PiperOrigin-RevId: 284566659
Diffstat (limited to 'mlir/lib/Transforms/LoopTiling.cpp')
0 files changed, 0 insertions, 0 deletions
