diff --git a/paddle/fluid/framework/new_executor/interpreter/stream_analyzer.cc b/paddle/fluid/framework/new_executor/interpreter/stream_analyzer.cc index 920fec72bd43ae577874ec5113a6e62d079af52b..88fac23338f543fa2470d3052e1d56dcdc549f5a 100644 --- 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_)) {