Use enumerate

openxla · Apr 29, 2024 · f34ea2e · f34ea2e
1 parent 24f35ac
commit f34ea2e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/stablehlo/dialect/TypeInference.cpp b/stablehlo/dialect/TypeInference.cpp
@@ -3717,8 +3717,7 @@ LogicalResult verifyDynamicBroadcastInDimOp(
   if (SmallVector<int64_t> shape;
       operandType.hasStaticShape() &&
       matchInts(outputDimensions, shape).succeeded()) {
-    for (int64_t i = 0; i != bcastDimensionsSize; ++i) {
-      auto dimIndex = broadcastDimensions[i];
+    for (auto [i, dimIndex] : llvm::enumerate(broadcastDimensions)) {
       if (!operandType.isDynamicDim(i)) {
         auto dimSize = operandType.getDimSize(i);
         auto shapeDimSize = shape[dimIndex];