diff --git a/csrc/cpu_ops.cpp b/csrc/cpu_ops.cpp index 6a31d9949..febd86e2e 100644 --- a/csrc/cpu_ops.cpp +++ b/csrc/cpu_ops.cpp @@ -323,7 +323,7 @@ void quantize_cpu_impl(float* code, const T* A, float* absmax, unsigned char* ou absmax[b] = absmax_block; if (absmax_block == 0.0f) { - std::fill(out + block_start, out + block_end, 0); + for (long long i = block_start; i < block_end; ++i) { out[i] = 0; } continue; }