[OPTIMIZER] Fixed up order of shared layouts (#881)

2022-11-21 06:25:02 +01:00
parent 4d64ffb5fe
commit 23f71daa27
6 changed files with 27 additions and 27 deletions
--- a/lib/Dialect/TritonGPU/Transforms/Pipeline.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/Pipeline.cpp
@@ -201,7 +201,9 @@ LogicalResult LoopPipeliner::initialize() {
                                             ty.getShape().end());
            bufferShape.insert(bufferShape.begin(), numStages);
            auto sharedEnc = ttg::SharedEncodingAttr::get(
-                ty.getContext(), dotOpEnc, ty.getShape(), ty.getElementType());
+                ty.getContext(), dotOpEnc, ty.getShape(),
+                triton::gpu::getOrder(ty.getEncoding()),
+                ty.getElementType());
            loadsBufferType[loadOp] = RankedTensorType::get(
                bufferShape, ty.getElementType(), sharedEnc);
          }