未验证 提交 0cbc870e 编写于 作者: R Ruibiao Chen 提交者: GitHub

Skip distributed Program for standalone executor (#44897)

上级 206102af
...@@ -1419,6 +1419,12 @@ class Executor(object): ...@@ -1419,6 +1419,12 @@ class Executor(object):
if program._build_strategy is not None and program._build_strategy.allow_cuda_graph_capture: if program._build_strategy is not None and program._build_strategy.allow_cuda_graph_capture:
return False return False
# Unsupported case 6: distributed
if program._build_strategy is not None and (
program._build_strategy.is_distribution
or program._build_strategy.num_trainers > 1):
return False
# Unsupported case 6 : disabled by FLAGS_CONVERT_GRAPH_TO_PROGRAM # Unsupported case 6 : disabled by FLAGS_CONVERT_GRAPH_TO_PROGRAM
if os.environ.get('FLAGS_CONVERT_GRAPH_TO_PROGRAM', if os.environ.get('FLAGS_CONVERT_GRAPH_TO_PROGRAM',
None) not in [1, '1', True, 'True', 'true']: None) not in [1, '1', True, 'True', 'true']:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册