[Intel] Fix build failures from 63cecbd

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

third_party/intel/lib/Analysis/DPAS.cpp

@@ -139,8 +139,8 @@ DPASAnalysis::getDPASType(OpTy op) {
   }
 
   if constexpr (std::is_same_v<OpTy, DotScaledOp>) {
-    aTy = cast<RankedTensorType>(op.getLhs().getType());
-    bTy = cast<RankedTensorType>(op.getRhs().getType());
+    aTy = cast<RankedTensorType>(op.getA().getType());
+    bTy = cast<RankedTensorType>(op.getB().getType());
     aElemTy = aTy.getElementType();
     bElemTy = bTy.getElementType();
 

third_party/intel/lib/TritonIntelGPUTransforms/AccelerateMatmul.cpp

@@ -214,16 +214,16 @@ class DecomposeScaledBlocked : public OpRewritePattern<tt::DotScaledOp> {
     if (rank == 3)
       return rewriter.notifyMatchFailure(scaledDotOp, "NYI: 3d case");
 
-    TensorValue a = scaledDotOp.getLhs();
-    TensorValue b = scaledDotOp.getRhs();
-    TensorValue aScale = scaledDotOp.getLhsScale();
-    TensorValue bScale = scaledDotOp.getRhsScale();
+    TensorValue a = scaledDotOp.getA();
+    TensorValue b = scaledDotOp.getB();
+    TensorValue aScale = scaledDotOp.getAScale();
+    TensorValue bScale = scaledDotOp.getBScale();
     if (aScale && bScale)
       return rewriter.notifyMatchFailure(scaledDotOp,
                                          "NYI: both LHS and RHS scale");
 
-    tt::ScaleDotElemType aElemType = scaledDotOp.getLhsType();
-    tt::ScaleDotElemType bElemType = scaledDotOp.getRhsType();
+    tt::ScaleDotElemType aElemType = scaledDotOp.getAElemType();
+    tt::ScaleDotElemType bElemType = scaledDotOp.getBElemType();
     auto supportsTypes = [](tt::ScaleDotElemType elemType) {
       return elemType == tt::ScaleDotElemType::E2M1 ||
              elemType == tt::ScaleDotElemType::E4M3 ||
@@ -363,10 +363,10 @@ class DecomposeScaledBlocked : public OpRewritePattern<tt::DotScaledOp> {
   getDPASEncoding(tt::DotScaledOp scaledDotOp,
                   PatternRewriter &rewriter) const {
     auto mod = scaledDotOp->getParentOfType<ModuleOp>();
-    TensorValue a = scaledDotOp.getLhs();
-    TensorValue b = scaledDotOp.getRhs();
-    TensorValue aScale = scaledDotOp.getLhsScale();
-    TensorValue bScale = scaledDotOp.getRhsScale();
+    TensorValue a = scaledDotOp.getA();
+    TensorValue b = scaledDotOp.getB();
+    TensorValue aScale = scaledDotOp.getAScale();
+    TensorValue bScale = scaledDotOp.getBScale();
     assert((!aScale || !bScale) && "NYI: both LHS and RHS scale");
 
     Type elemType =
@@ -613,22 +613,22 @@ static void sinkTransposeOp(tt::TransOp input) {
 }
 
 static tt::TransOp transposeDotOp(tt::DotScaledOp dotOp) {
-  assert(dotOp.getLhsScale() == nullptr && dotOp.getRhsScale() != nullptr &&
+  assert(dotOp.getAScale() == nullptr && dotOp.getBScale() != nullptr &&
          "Transpose DotOp expects scale on RHS");
   OpBuilder builder(dotOp);
-  Value lhs = dotOp.getLhs();
+  Value lhs = dotOp.getA();
   std::array<int, 2> transOrder = {1, 0};
   auto lhsTransposed =
       builder.create<tt::TransOp>(lhs.getLoc(), lhs, transOrder);
-  Value rhs = dotOp.getRhs();
+  Value rhs = dotOp.getB();
   auto rhsTransposed =
       builder.create<tt::TransOp>(rhs.getLoc(), rhs, transOrder);
   Value c = dotOp.getC();
   auto cTransposed = builder.create<tt::TransOp>(c.getLoc(), c, transOrder);
   auto result = builder.create<tt::DotScaledOp>(
       dotOp.getLoc(), cTransposed.getType(), rhsTransposed, lhsTransposed,
-      cTransposed, dotOp.getRhsScale(), dotOp.getLhsScale(), dotOp.getRhsType(),
-      dotOp.getLhsType(), dotOp.getFastMath());
+      cTransposed, dotOp.getBScale(), dotOp.getAScale(), dotOp.getBElemType(),
+      dotOp.getAElemType(), dotOp.getFastMath());
   auto transOp =
       builder.create<tt::TransOp>(result.getLoc(), result, transOrder);
   dotOp.replaceAllUsesWith(transOp.getOperation());
@@ -639,7 +639,7 @@ static tt::TransOp transposeDotOp(tt::DotScaledOp dotOp) {
 static void transposeDots(ModuleOp m) {
   SmallVector<tt::DotScaledOp> toTranspose;
   m.walk([&](tt::DotScaledOp dotOp) -> void {
-    if (dotOp.getLhsScale() == nullptr && dotOp.getRhsScale() != nullptr)
+    if (dotOp.getAScale() == nullptr && dotOp.getBScale() != nullptr)
       toTranspose.push_back(dotOp);
   });
   SmallVector<tt::TransOp> transposes;