为TensorMeta添加stop_gradient属性

Author: zty-king

python/paddle/distributed/auto_parallel/pipelining/utils.py

@@ -123,6 +123,7 @@ def __init__(self, tensor: paddle.Tensor):
             self._local_shape = None
         self.dtype = tensor.dtype
         self.placements = None if not tensor.is_dist() else tensor.placements
+        self.stop_gradient = tensor.stop_gradient
 
     def __repr__(self):
         return f"TensorMeta(global_shape={self.shape},local_shape={self._local_shape}, dtype={self.dtype}, placements={self.placements})"