[fluid_ops] Modify _legacy_C_ops.c_identity (#70299)

co63oc · web-flow · commit 288a2716ca5c · 2024-12-25T10:56:59.000+08:00
diff --git a/python/paddle/distributed/fleet/layers/mpu/mp_ops.py b/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):
