[DAS-RMSNorm] TE 2.3 returns 3 values of rmsnorm

API `rmsnorm_forward` of python returns 3 values rather than 2 from V2.3
Signed-off-by: wenjh <wenjh@sugon.com>

transformer_engine/pytorch/module/_common.py

@@ -88,8 +88,9 @@ def apply_normalization(
     normalization_func = _get_normalization_func(normalization, True)
 
     inputs = (inputmat, ln_weight) if ln_bias is None else (inputmat, ln_weight, ln_bias)
-    if enable_lightop and (ln_bias is None):
-        return rmsnorm_forward(inputmat, ln_weight,ln_out,eps,True)
+    if enable_lightop and (ln_bias is None) and normalization == "RMSNorm":
+        out, rsigma = rmsnorm_forward(inputmat, ln_weight,ln_out,eps,True)
+        return out, None, rsigma
     else:
         return normalization_func(
             *inputs,