change aclrtSynchronizeDevice to aclrtSynchronizeStream for better pe…

…rformance (PaddlePaddle#32196) * change aclrtSynchronizeDevice to aclrtSynchronizeStream for better performace * refine code
zhiqiu · Apr 15, 2021 · 62d42f0 · 62d42f0
1 parent 1ef2b93
commit 62d42f0
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
@@ -255,9 +255,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(); }