projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 3b46283) | patch
[mlir][gpu] Add op to create MMA constant matrix
authorthomasraoux <thomasraoux@google.com>
Wed, 9 Jun 2021 16:42:32 +0000 (09:42 -0700)
committerthomasraoux <thomasraoux@google.com>
Thu, 10 Jun 2021 15:34:04 +0000 (08:34 -0700)
commit428a62f65f16f1640b1bfe033d20e6a4f545dd3e
treec8b9ca5dcd2a8190902317fe24a1f4c69eaf8a62tree | snapshot
parent3b46283c1539f89619f2b40ab7732f434d7c68ffcommit | diff
[mlir][gpu] Add op to create MMA constant matrix

This allow creating a matrix with all elements set to a given value. This is
needed to be able to implement a simple dot op.

Differential Revision: https://reviews.llvm.org/D103870
mlir/include/mlir/Dialect/GPU/GPUOps.td diff | blob | history
mlir/lib/Conversion/GPUToNVVM/WmmaOpsToNvvm.cpp diff | blob | history
mlir/test/Conversion/GPUToNVVM/wmma-ops-to-nvvm.mlir diff | blob | history
mlir/test/Dialect/GPU/ops.mlir diff | blob | history
Domain: System / Toolchain;