[mlir][spirv] Add cooperative matrix store op
authorJakub Kuderski <kubak@google.com>
Wed, 19 Jul 2023 15:01:07 +0000 (11:01 -0400)
committerJakub Kuderski <kubak@google.com>
Wed, 19 Jul 2023 15:01:09 +0000 (11:01 -0400)
commit68cd1dbc2ec97e20306694a7cdc480584295e62c
treeb9fe63c20d44d1149d1c8fefa5fc5e1439e86bbb
parent1fa9e150b43dad347bd488bac584b3c2ed77d9f8
[mlir][spirv] Add cooperative matrix store op

Implement cooperative matrix store for the `SPV_KHR_cooperative_matrix`
extension: https://github.com/KhronosGroup/SPIRV-Registry/blob/master/extensions/KHR/SPV_KHR_cooperative_matrix.html.

Reviewed By: antiagainst

Differential Revision: https://reviews.llvm.org/D155631
mlir/include/mlir/Dialect/SPIRV/IR/SPIRVBase.td
mlir/include/mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td
mlir/lib/Dialect/SPIRV/IR/SPIRVOps.cpp
mlir/test/Dialect/SPIRV/IR/cooperative-matrix-ops.mlir