aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/Bitcode/Reader/BitcodeReader.cpp
diff options
context:
space:
mode:
authorgysit <gysit@google.com>2022-02-11 08:20:37 +0000
committergysit <gysit@google.com>2022-02-11 08:27:49 +0000
commita3655de2c81fc959590c109d81a010fc8e09c48e (patch)
tree89200935cbcf800f2adfab6833fa081d812cefd0 /llvm/lib/Bitcode/Reader/BitcodeReader.cpp
parentb21f497a782e155b19f41ec48da7ef75e8120195 (diff)
downloadllvm-a3655de2c81fc959590c109d81a010fc8e09c48e.zip
llvm-a3655de2c81fc959590c109d81a010fc8e09c48e.tar.gz
llvm-a3655de2c81fc959590c109d81a010fc8e09c48e.tar.bz2
[mlir][OpDSL] Add support for basic rank polymorphism.
Previously, OpDSL did not support rank polymorphism, which required a separate implementation of linalg.fill. This revision extends OpDSL to support rank polymorphism for a limited class of operations that access only scalars and tensors of rank zero. At operation instantiation time, it scales these scalar computations to multi-dimensional pointwise computations by replacing the empty indexing maps with identity index maps. The revision does not change the DSL itself, instead it adapts the Python emitter and the YAML generator to generate different indexing maps and and iterators depending on the rank of the first output. Additionally, the revision introduces a `linalg.fill_tensor` operation that in a future revision shall replace the current handwritten `linalg.fill` operation. `linalg.fill_tensor` is thus only temporarily available and will be renamed to `linalg.fill`. Reviewed By: nicolasvasilache, stellaraccident Differential Revision: https://reviews.llvm.org/D119003
Diffstat (limited to 'llvm/lib/Bitcode/Reader/BitcodeReader.cpp')
0 files changed, 0 insertions, 0 deletions