diff --git a/lib/Dialect/Torch/Transforms/ReduceOpVariants.cpp b/lib/Dialect/Torch/Transforms/ReduceOpVariants.cpp index bc4358831aa1..38d88e4a93ca 100644 --- a/lib/Dialect/Torch/Transforms/ReduceOpVariants.cpp +++ b/lib/Dialect/Torch/Transforms/ReduceOpVariants.cpp @@ -269,6 +269,9 @@ void TorchMatchSpecializedBackendOp::populateSpecializedConversions( llvm::SmallVector newOperands{ oldOperands[0], oldOperands[1], oldOperands[2], oldOperands[3], oldOperands[4], oldOperands[5], oldOperands[7]}; + Value enableGQA = + rewriter.create(op->getLoc(), false); + newOperands.push_back(enableGQA); auto newOp = rewriter.create( op.getLoc(), op->getResultTypes()[0], newOperands,