Work around non-standard std::uniform_int_distribution<uint8_t>
std::uniform_int_distribution<uint8_t> is apparently non-standard and
unsupported by MSVC
PiperOrigin-RevId: 306408109
diff --git a/bench/max-pooling.cc b/bench/max-pooling.cc
index 8d646f4..210f788 100644
--- a/bench/max-pooling.cc
+++ b/bench/max-pooling.cc
@@ -10,6 +10,7 @@
#include <cfloat>
#include <cmath>
#include <functional>
+#include <limits>
#include <random>
#include <vector>
@@ -30,7 +31,7 @@
std::random_device random_device;
auto rng = std::mt19937(random_device());
- auto u8rng = std::bind(std::uniform_int_distribution<uint8_t>(), rng);
+ auto u8rng = std::bind(std::uniform_int_distribution<uint32_t>(0, std::numeric_limits<uint8_t>::max()), rng);
const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1;
const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1;