comments

Signed-off-by: Chen Tong <[email protected]>
onnx · Feb 17, 2025 · 45447e7 · 45447e7
1 parent d0a2985
commit 45447e7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.cpp b/src/Conversion/ONNXToKrnl/ONNXToKrnlCommon.cpp
@@ -925,7 +925,9 @@ void genSafeCodeForGatherAlike(mlir::ConversionPatternRewriter &rewriter,
         IndexExpr index = NonAffineIndexExpr(indexVal);
 
         // index should be in range of [-r, r-1], where r = dim size of
-        // data[axis]
+        // data[axis].
+        // Assume that the index is loaded from tensor with negative value
+        // correction.
         Value errorCondition =
             ((index < (-1) * axisDim) | (index >= axisDim)).getValue();
         rewriter.create<scf::IfOp>(