diff options
Diffstat (limited to 'src/RefImplementations.cc')
-rw-r--r-- | src/RefImplementations.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/RefImplementations.cc b/src/RefImplementations.cc index 10e581f..6bf2d65 100644 --- a/src/RefImplementations.cc +++ b/src/RefImplementations.cc @@ -9,6 +9,7 @@ #include <cassert> #include <cmath> #include <cstring> +#include <algorithm> using namespace std; @@ -45,7 +46,7 @@ void requantize_u8acc32_ref( out[i * ld + j] = std::max( fuse_relu ? static_cast<int64_t>(C_zero_point) : 0l, - std::min(255l, rounded)); + std::min(static_cast<int64_t>(255l), rounded)); } } } |