fix selu

Tencent · Aug 27, 2024 · 7a3e9bd · 7a3e9bd
1 parent ad9d977
commit 7a3e9bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layer/riscv/selu_riscv.cpp b/src/layer/riscv/selu_riscv.cpp
@@ -46,8 +46,8 @@ int SELU_riscv::forward_inplace(Mat& bottom_top_blob, const Option& opt) const
 
             _p = __riscv_vfmul_vf_f32m8_mu(_higher, _p, _p, lambda, vl);
             vfloat32m8_t _nps = exp_ps(_p, vl);
-            _nps = __riscv_vfsub_vf_f32m8_mu(_lower, _nps, _nps, 1.f, vl);
-            _nps = __riscv_vfmul_vf_f32m8_mu(_lower, _nps, _nps, alphaxlambda, vl);
+            _nps = __riscv_vfsub_vf_f32m8_mu(_lower, _p, _nps, 1.f, vl);
+            _nps = __riscv_vfmul_vf_f32m8_mu(_lower, _p, _nps, alphaxlambda, vl);
 
             __riscv_vse32_v_f32m8(ptr, _nps, vl);
             ptr += vl;