Fix overflow in prediction size. (#7885)

2022-05-12 02:44:03 +08:00
parent 8ba4722d04
commit 94ca52b7b7
2 changed files with 4 additions and 2 deletions
--- a/src/c_api/c_api_utils.h
+++ b/src/c_api/c_api_utils.h
@@ -115,7 +115,7 @@ inline void CalcPredictShape(bool strict_shape, PredictionType type, size_t rows
  }
  }
  CHECK_EQ(
-      std::accumulate(shape.cbegin(), shape.cend(), 1, std::multiplies<>{}),
+      std::accumulate(shape.cbegin(), shape.cend(), static_cast<bst_ulong>(1), std::multiplies<>{}),
      chunksize * rows);
 }