未验证 提交 46c04c8f 编写于 作者: W Weilong Wu 提交者: GitHub

Fix format for doc (#43341)

上级 c43a1ff6
...@@ -265,16 +265,16 @@ def ipu_shard_guard(index=None, stage=None): ...@@ -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. Used to shard the graph on IPUs. Set each Op run on which IPU in the sharding and which stage in the pipelining.
Args: 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. 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, 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. which means no pipelining computation order and run Ops in terms of graph.
**Note**: **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` . 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` . 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 A index is allowed to match none stage or a stage. A stage is only allowed to match a new or
duplicated index. duplicated index.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册