Fix

Author: co63oc

python/paddle/distributed/fleet/layers/mpu/mp_ops.py

@@ -44,15 +44,7 @@ def forward(ctx, tensor, group, skip_c_identity_dynamic):
         if skip_c_identity_dynamic:
             return tensor
         else:
-            return _legacy_C_ops.c_identity(
-                tensor,
-                'use_calc_stream',
-                True,
-                'ring_id',
-                group.id,
-                'use_model_parallel',
-                True,
-            )
+            return _C_ops.c_identity(tensor, group.id, True, True)
 
     @staticmethod
     def backward(ctx, dy):