fix: fix rms_norm params (#18)

Author: yao-fengchen

lmdeploy/pytorch/kernels/ascend/rms_norm.py

@@ -3,5 +3,13 @@
 from torch import Tensor
 
 
-def rms_norm(hidden_states: Tensor, weight: Tensor, epsilon: float = 1e-6):
-    return ext_ops.rms_norm(hidden_states, weight, epsilon)
+def rms_norm(hidden_states: Tensor,
+             weight: Tensor,
+             eps: float = 1e-6,
+             out: Tensor = None):
+    rms_norm_out = ext_ops.rms_norm(hidden_states, weight, eps)
+    if out is None:
+        out = rms_norm_out
+    else:
+        out.copy_(rms_norm_out)
+    return rms_norm_out