[mlir][spirv] First step to support spirv cooperative matrix extension.
authorThomas Raoux <thomasraoux@google.com>
Wed, 20 May 2020 02:07:21 +0000 (19:07 -0700)
committerThomas Raoux <thomasraoux@google.com>
Wed, 20 May 2020 02:29:41 +0000 (19:29 -0700)
commitb359bbaa8b41a84ae54369e3017ce1a5c7afe1a1
tree49c25a8f8cac9159984f921bf9280baf03cfb100
parent2b59e9f1bdd8d0f8a315f9ddba2cdc87a3e682fb
[mlir][spirv] First step to support spirv cooperative matrix extension.

Add a new type to SPIRV dialect for cooperative matrix and add new op for
cooperative matrix load. This is missing most instructions to support
cooperative matrix extension but this is a stop-gap patch to avoid creating big
review.

Differential Revision: https://reviews.llvm.org/D80043
13 files changed:
mlir/include/mlir/Dialect/SPIRV/ParserUtils.h [new file with mode: 0644]
mlir/include/mlir/Dialect/SPIRV/SPIRVBase.td
mlir/include/mlir/Dialect/SPIRV/SPIRVCooperativeMatrixOps.td [new file with mode: 0644]
mlir/include/mlir/Dialect/SPIRV/SPIRVOps.td
mlir/include/mlir/Dialect/SPIRV/SPIRVTypes.h
mlir/lib/Dialect/SPIRV/SPIRVDialect.cpp
mlir/lib/Dialect/SPIRV/SPIRVOps.cpp
mlir/lib/Dialect/SPIRV/SPIRVTypes.cpp
mlir/lib/Dialect/SPIRV/Serialization/Deserializer.cpp
mlir/lib/Dialect/SPIRV/Serialization/Serializer.cpp
mlir/test/Dialect/SPIRV/Serialization/cooperative-matrix.mlir [new file with mode: 0644]
mlir/test/Dialect/SPIRV/cooperative-matrix.mlir [new file with mode: 0644]
mlir/test/Dialect/SPIRV/types.mlir