change aclrtSynchronizeDevice to aclrtSynchronizeStream for better pe…

…rformance (#32196) * change aclrtSynchronizeDevice to aclrtSynchronizeStream for better performace * refine code
PaddlePaddle · Apr 12, 2021 · 054f8e7 · 054f8e7 · paddle-bot-old · Apr 12, 2021
1 parent 7039c06
commit 054f8e7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/paddle/fluid/platform/device_context.cc b/paddle/fluid/platform/device_context.cc
@@ -254,9 +254,8 @@ NPUDeviceContext::~NPUDeviceContext() {
 
 void NPUDeviceContext::Wait() const {
   platform::RecordEvent record_event("NPUDeviceContext/wait");
-  NPUDeviceGuard guard(place_.device);
   VLOG(4) << "NPU context Wait";
-  PADDLE_ENFORCE_NPU_SUCCESS(aclrtSynchronizeDevice());
+  stream_->Wait();
 }
 
 aclrtStream NPUDeviceContext::stream() const { return stream_->raw_stream(); }