Update comments and implementation.

Signed-off-by: hanhanW <[email protected]>

Author: hanhanW

mlir/include/mlir/Dialect/Utils/StaticValueUtils.h

@@ -24,12 +24,10 @@
 
 namespace mlir {
 
-/// Return true if `v` is an IntegerAttr with value `0` of a ConstantIndexOp
-/// with attribute with value `0`.
+/// Return true if `v` is an IntegerAttr with value `0`.
 bool isZeroIndex(OpFoldResult v);
 
-/// Return true if `v` is an IntegerAttr with value `1` of a ConstantIndexOp
-/// with attribute with value `1`.
+/// Return true if `v` is an IntegerAttr with value `1`.
 bool isOneIndex(OpFoldResult v);
 
 /// Represents a range (offset, size, and stride) where each element of the

mlir/lib/Dialect/Utils/StaticValueUtils.cpp

@@ -15,17 +15,9 @@
 
 namespace mlir {
 
-bool isZeroIndex(OpFoldResult v) {
-  if (!v)
-    return false;
-  return isConstantIntValue(v, 0);
-}
+bool isZeroIndex(OpFoldResult v) { return isConstantIntValue(v, 0); }
 
-bool isOneIndex(OpFoldResult v) {
-  if (!v)
-    return false;
-  return isConstantIntValue(v, 1);
-}
+bool isOneIndex(OpFoldResult v) { return isConstantIntValue(v, 1); }
 
 std::tuple<SmallVector<OpFoldResult>, SmallVector<OpFoldResult>,
            SmallVector<OpFoldResult>>