From 7392578d7a26672c8f97f0edc60d673d6f1c7f5b Mon Sep 17 00:00:00 2001 From: Leo Chen Date: Fri, 11 Feb 2022 12:44:25 +0800 Subject: [PATCH] [new-exec] set type of op-kernel op by place (#39458) --- .../framework/new_executor/interpretercore_util.cc | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/new_executor/interpretercore_util.cc b/paddle/fluid/framework/new_executor/interpretercore_util.cc index 50725e5eea5..7a0de437572 100644 --- a/paddle/fluid/framework/new_executor/interpretercore_util.cc +++ b/paddle/fluid/framework/new_executor/interpretercore_util.cc @@ -263,7 +263,15 @@ void deal_operator_base(const platform::Place& place, auto* dev_ctx = pool.Get(place); // input, output is prepared. set the other attributes. op_func_node->operator_base_ = op_base; - op_func_node->type_ = OpFuncType::kQueueSync; // alway Sync + if (platform::is_gpu_place(place)) { + op_func_node->type_ = OpFuncType::kQueueAsync; + } else if (platform::is_cpu_place(place)) { + op_func_node->type_ = OpFuncType::kQueueSync; + } else { + PADDLE_THROW( + platform::errors::Fatal("Unsupported current place %s", place)); + } + op_func_node->kernel_func_ = nullptr; op_base->Run(*local_scope, place); // Run without data transformer. -- GitLab