From e9b92018ce7f84d50d7408bc02f1801d28fec442 Mon Sep 17 00:00:00 2001 From: Ruibiao Chen Date: Thu, 28 Jul 2022 20:10:37 +0800 Subject: [PATCH] Skip CUDA Graph case for standalone executor (#44693) --- python/paddle/fluid/executor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/executor.py b/python/paddle/fluid/executor.py index 867b40c1fe0..bbdf6002279 100755 --- a/python/paddle/fluid/executor.py +++ b/python/paddle/fluid/executor.py @@ -1413,7 +1413,11 @@ class Executor(object): if program._is_inference: 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', None) not in [1, '1', True, 'True', 'true']: return False -- GitLab