Align TritonGPU/IR/TritonGPUAttrDefs.td with upstream (#5173)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

include/triton/Dialect/TritonGPU/IR/TritonGPUAttrDefs.td

@@ -64,8 +64,8 @@ def LinearLayoutParam : AttrOrTypeParameter<"LinearLayout",
 // Base Attribute
 //===----------------------------------------------------------------------===//
 
-class TritonGPU_Attr<string name, string attrMnemonic, list<Trait> traits = [], Dialect dialect = TritonGPU_Dialect>
-  : AttrDef<dialect, name, traits> {
+class TritonGPU_Attr<string name, string attrMnemonic, list<Trait> traits = []>
+  : AttrDef<TritonGPU_Dialect, name, traits> {
 
   let description = [{
 TritonGPU tensors differ from usual tensors in that they contain a _layout_ attribute which determines
@@ -740,9 +740,8 @@ We call each individual tile "rep".
   ];
 }
 
-class DistributedEncoding<string name, string attrMnemonic, list<Trait> traits = [],
-                     Dialect dialect = TritonGPU_Dialect>
-  : TritonGPU_Attr<name, attrMnemonic, !listconcat([DistributedEncodingTrait, LayoutEncodingTrait], traits), dialect> {
+class DistributedEncoding<string name, string attrMnemonic, list<Trait> traits = []>
+  : TritonGPU_Attr<name, attrMnemonic, !listconcat([DistributedEncodingTrait, LayoutEncodingTrait], traits)> {
 
   let description = [{
 Distributed encodings have a layout function L that is entirely characterized

third_party/intel/include/Dialect/TritonIntelGPU/IR/TritonIntelGPUAttrDefs.td

@@ -10,7 +10,8 @@ include "triton/Dialect/TritonGPU/IR/TritonGPUAttrDefs.td"
 //===----------------------------------------------------------------------===//
 
 def DpasEncodingAttr : DistributedEncoding<"DpasEncoding", "intel_dpas_encoding",
-                                          [MmaEncodingTrait, DeclareLayoutEncodingMethods], TritonIntelGPU_Dialect> {
+                                          [MmaEncodingTrait, DeclareLayoutEncodingMethods]> {
+  let dialect = TritonIntelGPU_Dialect;
   let mnemonic = "dpas";
 
   let description = [{
@@ -254,7 +255,8 @@ The semantic of this `tt.dot` includes GEMM tiling configuration as:
 //===----------------------------------------------------------------------===//
 
 def WarpEncodingAttr : DistributedEncoding<"WarpEncoding", "intel_warp_encoding",
-                                     [DeclareLayoutEncodingMethods], TritonIntelGPU_Dialect> {
+                                     [DeclareLayoutEncodingMethods]> {
+  let dialect = TritonIntelGPU_Dialect;
   let mnemonic = "warp";
 
   let description = [{
@@ -284,7 +286,8 @@ def WarpEncodingAttr : DistributedEncoding<"WarpEncoding", "intel_warp_encoding"
 // Intel Subgroup2DBlock Encoding
 //===----------------------------------------------------------------------===//
 
-def Subgroup2DBlockEncodingAttr : DistributedEncoding<"Subgroup2DBlockEncoding", "subgroup_2d_block_encoding", [MmaEncodingTrait], TritonIntelGPU_Dialect> {
+def Subgroup2DBlockEncodingAttr : DistributedEncoding<"Subgroup2DBlockEncoding", "subgroup_2d_block_encoding", [MmaEncodingTrait]> {
+  let dialect = TritonIntelGPU_Dialect;
   let mnemonic = "subgroup_2d_block";
 
   let description = [{