llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix comments

pre_fixup_20240110
Frederik Harwath 2024-01-10 07:26:12 -08:00 committed by Frederik Harwath
parent de85278f1f
commit 2230c0de90
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Conversion/TorchToLinalg/DataMovement.cpp
View File

@ -491,7 +491,7 @@ public:
loc, input, extractOffsets, extractShape, allOneStrides); loc, input, extractOffsets, extractShape, allOneStrides);
// Reverse the tile along the horizontal, vertical, or both // Reverse the tile along the horizontal, vertical, or both
// dimensions // dimensions.
auto inputMap = AffineMap::getMultiDimIdentityMap(numDims, context); auto inputMap = AffineMap::getMultiDimIdentityMap(numDims, context);
if (shouldHReflect(horizontalPos)) { if (shouldHReflect(horizontalPos)) {
inputMap = inputMap =
@ -511,7 +511,7 @@ public:
}) })
.getResult(0); .getResult(0);
// Insert the tile in the resultTensor // Insert the tile in the resultTensor.
SmallVector<Value> insertOffsets(numDims, zero); SmallVector<Value> insertOffsets(numDims, zero);
insertOffsets[hDim] = insertHOffset[horizontalPos]; insertOffsets[hDim] = insertHOffset[horizontalPos];
insertOffsets[vDim] = insertVOffset[verticalPos]; insertOffsets[vDim] = insertVOffset[verticalPos];