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

Skip CUDA Graph case for standalone executor (#44693)

上级 8aa286dd
...@@ -1413,7 +1413,11 @@ class Executor(object): ...@@ -1413,7 +1413,11 @@ class Executor(object):
if program._is_inference: if program._is_inference:
return False return False
# Unsupported case 5 : disabled by FLAGS_CONVERT_GRAPH_TO_PROGRAM # Unsupported case 5: CUDA Graph
if program._build_strategy is not None and program._build_strategy.allow_cuda_graph_capture:
return False
# 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']:
return False return False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册