add annotation

PaddlePaddle · Jan 10, 2024 · fe580a3 · fe580a3
1 parent 0b5470a
commit fe580a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/paddle/distributed/auto_parallel/api.py b/python/paddle/distributed/auto_parallel/api.py
@@ -228,6 +228,8 @@ def _init_func(var, block):
             dist_tensor = paddle.Tensor(
                 tensor, process_mesh=mesh, placements=placements, place=place
             )
+            # InitDistTensorWithTensor won't pass the stop gradient attribute,
+            # have to pass it manually.
             dist_tensor.stop_gradient = tensor.stop_gradient
             return dist_tensor
     else: