File tree Expand file tree Collapse file tree 2 files changed +2
-14
lines changed
include/Dialect/TritonIntelGPU/IR
lib/Dialect/TritonIntelGPU/IR Expand file tree Collapse file tree 2 files changed +2
-14
lines changed Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ include "triton/Dialect/TritonGPU/IR/TritonGPUAttrDefs.td"
1010//===----------------------------------------------------------------------===//
1111
1212def DpasEncodingAttr : DistributedEncoding<"DpasEncoding", "intel_dpas_encoding",
13- [MmaEncodingTrait], TritonIntelGPU_Dialect> {
13+ [MmaEncodingTrait, DeclareLayoutEncodingMethods ], TritonIntelGPU_Dialect> {
1414 let mnemonic = "dpas";
1515
1616 let description = [{
@@ -254,7 +254,7 @@ The semantic of this `tt.dot` includes GEMM tiling configuration as:
254254//===----------------------------------------------------------------------===//
255255
256256def WarpEncodingAttr : DistributedEncoding<"WarpEncoding", "intel_warp_encoding",
257- [], TritonIntelGPU_Dialect> {
257+ [DeclareLayoutEncodingMethods ], TritonIntelGPU_Dialect> {
258258 let mnemonic = "warp";
259259
260260 let description = [{
Original file line number Diff line number Diff line change @@ -635,18 +635,6 @@ SmallVector<unsigned> Subgroup2DBlockEncodingAttr::getRepOrder() const {
635635 return getMatrixOrder (getRank (), /* rowMajor*/ true );
636636}
637637
638- SmallVector<unsigned > Subgroup2DBlockEncodingAttr::getCTAsPerCGA () const {
639- return SmallVector<unsigned >(getCTALayout ().getCTAsPerCGA ());
640- }
641-
642- SmallVector<unsigned > Subgroup2DBlockEncodingAttr::getCTAOrder () const {
643- return SmallVector<unsigned >(getCTALayout ().getCTAOrder ());
644- }
645-
646- SmallVector<unsigned > Subgroup2DBlockEncodingAttr::getCTASplitNum () const {
647- return SmallVector<unsigned >(getCTALayout ().getCTASplitNum ());
648- }
649-
650638SmallVector<unsigned >
651639Subgroup2DBlockEncodingAttr::getRepOrderForOperand (int opIdx) const {
652640 return getOrderForDotOperand (opIdx, getRank (), /* kContig*/ true );
You can’t perform that action at this time.
0 commit comments