Set WaiterType of kGpuSync to kCPU (#48758)

a5999d83 · Ruibiao Chen · GitHub · c8497414 · a5999d83
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/framework/new_executor/interpreter/stream_analyzer.cc ...uid/framework/new_executor/interpreter/stream_analyzer.cc +2 -1

未找到文件。
--- a/paddle/fluid/framework/new_executor/interpreter/stream_analyzer.cc
+++ b/paddle/fluid/framework/new_executor/interpreter/stream_analyzer.cc
@@ -356,7 +356,8 @@ void StreamAnalyzer::ShrinkEventInfo(

 platform::DeviceType StreamAnalyzer::GetWaiterType(
    const Instruction& instr) const {
-  if (instr.KernelType() == OpFuncType::kCpuSync) {
+  if (instr.KernelType() == OpFuncType::kCpuSync ||
+      instr.KernelType() == OpFuncType::kGpuSync) {
    return platform::kCPU;
  } else {
    if (platform::is_xpu_place(place_)) {