diff --git a/lib/gc/Dialect/Linalgx/Utils.cpp b/lib/gc/Dialect/Linalgx/Utils.cpp index fe9096fe7..b918b0ec2 100644 --- a/lib/gc/Dialect/Linalgx/Utils.cpp +++ b/lib/gc/Dialect/Linalgx/Utils.cpp @@ -492,8 +492,8 @@ bool isGenericPackedMatmulOpImpl(linalg::GenericOp genericOp, return false; } // Check for packing - ValueRange inputs = genericOp.getDpsInputs(); - ValueRange outputs = genericOp.getDpsInits(); + ValueRange inputs = genericOp.getInputs(); + ValueRange outputs = genericOp.getOutputs(); auto shapeA = cast(inputs.front().getType()); auto shapeB = cast(inputs.back().getType()); auto shapeC = cast(outputs.back().getType());