From e3936dca933893a8849195989db8b9e5a0893316 Mon Sep 17 00:00:00 2001 From: Han-Chung Wang Date: Fri, 23 Aug 2024 11:15:29 -0700 Subject: [PATCH] [GPU][NFC] Update comments for MMAAttr attribute (#18339) The revision redirects the layout comment to the enum file which reduces the maintenance burden. People wont miss to update comments like this. IMO, it is also the naming convention, so we don't need to spell it loud. The revision also make the "section comment" consistent. --- .../compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td b/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td index c3514d6b8fbd..8222117213b5 100644 --- a/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td +++ b/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td @@ -172,6 +172,7 @@ class IREEGPU_MmaVectorLayoutAttr : //===----------------------------------------------------------------------===// // MMA intrinsic +//===----------------------------------------------------------------------===// class IREEGPU_MmaEnumAttr : EnumAttr; @@ -186,12 +187,8 @@ def IREEGPU_MMAAttr : IREEGPU_MmaVectorLayoutAttr<"MMA", "MMAIntrinsicAttr"> { let description = !strconcat(baseDescription, [{ This mma variant describes configurations for MMA ops. The |intrinsic| field specifies which particular MMA intrinsic this refers to, with each - intrinsic implicating a specific MNK shape and operand types. The intrinsic - enum name describes these fields as - - _MxNxK_ - - Where the element type for the `A` and `B` matrices are both `InputType`. + intrinsic implicating a specific MNK shape and operand types. + See IREEGPUEnums.td for the definition of the intrinsics. }]); let hasCustomAssemblyFormat = 1; @@ -230,6 +227,7 @@ def IREEGPU_MMAOpsArrayAttr : ArrayOfAttr< //===----------------------------------------------------------------------===// // MMA schedule +//===----------------------------------------------------------------------===// def IREEGPU_MmaScheduleAttr : AttrDef { let mnemonic = "mma_schedule"; @@ -259,6 +257,7 @@ def IREEGPU_MmaScheduleAttr : AttrDef { //===----------------------------------------------------------------------===// // Workgroup processor level description +//===----------------------------------------------------------------------===// def IREEGPU_TargetWgpAttr : AttrDef { let summary = "Workgroup processor level target description";