未验证 提交 ce733495 编写于 作者: B Baibaifan 提交者: GitHub

del wait in sharding for npu (#34637)

上级 c91b1e03
...@@ -379,8 +379,9 @@ class ShardingOptimizer(MetaOptimizerBase): ...@@ -379,8 +379,9 @@ class ShardingOptimizer(MetaOptimizerBase):
'w') as f: 'w') as f:
f.writelines(str(main_block.program)) f.writelines(str(main_block.program))
# GPU and NPU need to wait server ready # GPU need to wait server ready, GPU and NPU is Layered connection
self._wait() if not core.is_compiled_with_npu():
self._wait()
return optimize_ops, params_grads return optimize_ops, params_grads
def _init_pair_comm(self, pair, ring_id): def _init_pair_comm(self, pair, ring_id):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册