diff --git a/paddle/fluid/eager/backward.cc b/paddle/fluid/eager/backward.cc index a220fe18fb35d3fb39e5983fe9846892b3ecae0c..2216b6b01427eeb1778583b8725d4799beb94c71 100644 --- a/paddle/fluid/eager/backward.cc +++ b/paddle/fluid/eager/backward.cc @@ -113,7 +113,6 @@ std::vector RunBackward( std::queue force_sequential_nodes_forward_queue = egr::Controller::Instance().GetForceSequentialNodes(); - egr::Controller::Instance().ClearForceSequentialNodes(); std::deque force_sequential_nodes_queue; std::set force_sequential_nodes_set; std::set ready_force_sequential_nodes; @@ -421,6 +420,7 @@ void Backward(const std::vector& tensors, // outputs VLOG(3) << "Run in Backward"; paddle::platform::RecordEvent backward_record_event( "backward", paddle::platform::TracerEventType::UserDefined, 1); + egr::Controller::Instance().ClearForceSequentialNodes(); RunBackward(tensors, grad_tensors, retain_graph); phi::autotune::AutoTuneStatus::Instance().Update(); }