diff --git a/paddle/contrib/inference/paddle_inference_api_anakin_engine.cc b/paddle/contrib/inference/paddle_inference_api_anakin_engine.cc index ba2d30314715a57c5ab85e5ae1d8ac0512bbc74f..daf0b56825498d48ca48d1d5b0ea6a3d7314bfc4 100644 --- a/paddle/contrib/inference/paddle_inference_api_anakin_engine.cc +++ b/paddle/contrib/inference/paddle_inference_api_anakin_engine.cc @@ -54,6 +54,7 @@ bool PaddleInferenceAnakinPredictor::Run( LOG(ERROR) << "copy data from CPU to GPU error"; return false; } + cudaStreamSynchronize(NULL); } executor_.prediction(); @@ -76,6 +77,7 @@ bool PaddleInferenceAnakinPredictor::Run( LOG(ERROR) << "copy data from GPU to CPU error"; return false; } + cudaStreamSynchronize(NULL); } return true; }