diff --git a/lite/kernels/arm/scale_compute.cc b/lite/kernels/arm/scale_compute.cc index 7f9949676e3..0076aa75d3d 100644 --- a/lite/kernels/arm/scale_compute.cc +++ b/lite/kernels/arm/scale_compute.cc @@ -31,7 +31,7 @@ void ScaleCompute::Run() { if (!param.bias_after_scale) { bias *= scale; } - if (std::isinf(abs(scale))) { + if (std::isinf(std::abs(scale))) { scale = 0; } T alpha = param.alpha;