stage

mlir/include/mlir/Dialect/XeGPU/IR/XeGPUDialect.td

@@ -38,6 +38,8 @@ def XeGPU_Dialect : Dialect {
     let useDefaultAttributePrinterParser = true;
 
     let extraClassDeclaration = [{
+      static constexpr const char *operandLayoutNamePrefix = "layout_operand_";
+      static constexpr const char *resultLayoutNamePrefix = "layout_result_";
       /// Checks if the given shape can be evenly distributed based on the layout
       /// and data factors provided by the LayoutAttr.
       static bool isEvenlyDistributable(llvm::ArrayRef<int64_t> shape, xegpu::LayoutAttr attr);

mlir/lib/Dialect/XeGPU/Transforms/XeGPUUnroll.cpp

@@ -378,7 +378,22 @@ struct UnrollDpasOp : public UnrollPattern<xegpu::DpasOp> {
   using UnrollPattern<xegpu::DpasOp>::UnrollPattern;
   LogicalResult matchAndRewrite(xegpu::DpasOp op,
                                 PatternRewriter &rewriter) const override {
+
+    auto loc = op.getLoc();
+
+    // a vector of 3 elements should be returned, representing M, K, N respectively.
+    auto maybeTargetShape = getTargetShape(options, op);
+    if (!maybeTargetShape || maybeTargetShape->size() != 3)
+      return failure();
+    auto M = (*maybeTargetShape)[0];
+    auto K = (*maybeTargetShape)[1];
+    auto N = (*maybeTargetShape)[2];
+
+    llvm::dbgs() << "\nM: " << M << ", K: " << K << ", N: " << N << "\n";
+
     return failure();
+
+
   }
 };