Correct dims for plugin interface

nod-ai · Sep 27, 2023 · f18fa8e · f18fa8e
1 parent 6847507
commit f18fa8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPULowerToAccelUKernels.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPULowerToAccelUKernels.cpp
@@ -140,8 +140,8 @@ matchDAGForUKernel(RewriterBase &rewriter, linalg::Mmt4DOp op) {
     Value k = getDimAsI32(rewriter, loc, reducedRhs, 1);
   */
   Value m = rewriter.create<tensor::DimOp>(loc, reducedLhs, 0);
-  Value n = rewriter.create<tensor::DimOp>(loc, reducedRhs, 0);
-  Value k = rewriter.create<tensor::DimOp>(loc, reducedRhs, 1);
+  Value n = rewriter.create<tensor::DimOp>(loc, reducedLhs, 1);
+  Value k = rewriter.create<tensor::DimOp>(loc, reducedRhs, 0);
   Value flagsVal = rewriter.create<arith::ConstantOp>(
       loc, rewriter.getI32IntegerAttr(flags));
   auto targetAttr = IREE::HAL::ExecutableTargetAttr::lookup(op);