diff --git a/lib/Dialect/TensorAlgebra/Transforms/TCtoTTGT.cpp b/lib/Dialect/TensorAlgebra/Transforms/TCtoTTGT.cpp index c383d1cd..1b4072c1 100644 --- a/lib/Dialect/TensorAlgebra/Transforms/TCtoTTGT.cpp +++ b/lib/Dialect/TensorAlgebra/Transforms/TCtoTTGT.cpp @@ -421,8 +421,11 @@ namespace MemRefType::get(lhsDims, lhsMemrefType.getElementType()), loc, rewriter); useLHSTranspose = true; + Value constantOp = rewriter.create(loc, rewriter.getF64FloatAttr(0.0)); + + rewriter.create(loc, constantOp, lhsAlloc); // TODO(gkestor): we might need this copy if we support update C[] += A[] * B[] - rewriter.create(loc, lhsMemref, lhsAlloc, llvm::ArrayRef(lhsOutPerm_int64)); + // rewriter.create(loc, lhsMemref, lhsAlloc, llvm::ArrayRef(lhsOutPerm_int64)); } RankedTensorType collapsedTensorType;