diff --git a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td --- a/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td +++ b/mlir/include/mlir/Dialect/SPIRV/IR/SPIRVCooperativeMatrixOps.td @@ -39,7 +39,7 @@ For example: ``` - %0 = spirv.NV.CooperativeMatrixLength : !spirv.NV.coopmatrix + %0 = spirv.NV.CooperativeMatrixLength : !spirv.NV.coopmatrix<8x16xi32, Subgroup> ``` }]; @@ -115,7 +115,7 @@ ``` %0 = spirv.NV.CooperativeMatrixLoad %ptr, %stride, %colMajor - : !spirv.ptr as !spirv.NV.coopmatrix + : !spirv.ptr as !spirv.NV.coopmatrix<16x8xi32, Workgroup> ``` }]; @@ -186,7 +186,7 @@ ``` %0 = spirv.NV.CooperativeMatrixMulAdd %arg0, %arg1, %arg2, : - !spirv.NV.coopmatrix + !spirv.NV.coopmatrix<8x16xi32, Subgroup> ``` }]; @@ -251,7 +251,7 @@ ``` spirv.NV.CooperativeMatrixStore %arg0, %arg2, %arg1, %arg3 : - !spirv.ptr, !spirv.NV.coopmatrix + !spirv.ptr, !spirv.NV.coopmatrix<16x8xi32, Workgroup> ``` }];