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