diff --git a/lib/Conversion/TorchToLinalg/Linear.cpp b/lib/Conversion/TorchToLinalg/Linear.cpp index c9e92b625..9e585d20e 100644 --- a/lib/Conversion/TorchToLinalg/Linear.cpp +++ b/lib/Conversion/TorchToLinalg/Linear.cpp @@ -572,11 +572,11 @@ public: createZeroInitTensor(rewriter, loc, weightInitDims, elementType); SmallVector iteratorTypes(inRank, getParallelIteratorTypeName()); - SmallVector indexingMaps( - 2, AffineMap::getMultiDimIdentityMap(inRank, context)); + SmallVector indexingMaps{ + AffineMap::getMultiDimIdentityMap(inRank, context)}; weight = rewriter .create( - loc, weightInitTensor.getType(), weight, + loc, weightInitTensor.getType(), ValueRange{}, weightInitTensor, indexingMaps, iteratorTypes, [&](OpBuilder &b, Location loc, ValueRange args) { SmallVector indices;