diff --git a/sgl-kernel/csrc/cpu/activation.cpp b/sgl-kernel/csrc/cpu/activation.cpp index e1011bc49..da0951498 100644 --- a/sgl-kernel/csrc/cpu/activation.cpp +++ b/sgl-kernel/csrc/cpu/activation.cpp @@ -121,7 +121,7 @@ at::Tensor silu_and_mul_cpu(at::Tensor& input) { num_tokens, d, [](float x) { return x / (1.f + std::exp(-x)); }, - [](Vec x) { return x / (Vec(1.f) + x.neg().exp()); }); + [](Vec x) { return x / (Vec(1.f) + x.neg().exp_u20()); }); }); return out; }