diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index fdd5c0b47b4dc491e4278381621d9bde9f31b8d6..a790feb9af29c7af58fb99b673153ce861674502 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -265,16 +265,16 @@ def ipu_shard_guard(index=None, stage=None): Used to shard the graph on IPUs. Set each Op run on which IPU in the sharding and which stage in the pipelining. Args: - index(int, optional): Specify which ipu the Tensor is computed on, (such as ‘0, 1, 2, 3’). + index(int, optional): Specify which ipu the Tensor is computed on, (such as '0, 1, 2, 3'). The default value is None, which means the Op only run on IPU 0. - stage(int, optional): Specify the computation order of the sharded model(such as ‘0, 1, 2, 3’). + stage(int, optional): Specify the computation order of the sharded model(such as '0, 1, 2, 3'). The sharded model will be computed from small to large. The default value is None, which means no pipelining computation order and run Ops in terms of graph. **Note**: - Only if the enable_manual_shard=True, the ‘index’ is able to be set not None. Please refer + Only if the enable_manual_shard=True, the 'index' is able to be set not None. Please refer to :code:`paddle.static.IpuStrategy` . - Only if the enable_pipelining=True, the ‘stage’ is able to be set not None. Please refer + Only if the enable_pipelining=True, the 'stage' is able to be set not None. Please refer to :code:`paddle.static.IpuStrategy` . A index is allowed to match none stage or a stage. A stage is only allowed to match a new or duplicated index.