diff --git a/src/infiniop/ops/relu6/cuda/kernel.cuh b/src/infiniop/ops/relu6/cuda/kernel.cuh index 4072be2af..32676e304 100644 --- a/src/infiniop/ops/relu6/cuda/kernel.cuh +++ b/src/infiniop/ops/relu6/cuda/kernel.cuh @@ -21,7 +21,7 @@ struct Relu6Op { } else if constexpr (std::is_same_v) { return fminf(fmaxf(x, 0.0f), 6.0f); } else if constexpr (std::is_same_v) { - return std::min(std::max(x, 0.0), 6.0); + return fmin(fmax(x, 0.0), 6.0); } else { float xf = static_cast(x); float result = fminf(fmaxf(xf, 0.0f), 6.0f);