change aclrtSynchronizeDevice to aclrtSynchronizeStream for better performance (#32196)

zhiqiu · web-flow · commit 054f8e7a32b4 · 2021-04-12T15:21:21.000+08:00
* change aclrtSynchronizeDevice to aclrtSynchronizeStream for better performace

* refine code
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(); }

Original file line number	Diff line number	Diff line change
`@@ -254,9 +254,8 @@ NPUDeviceContext::~NPUDeviceContext() {`
`254`	`254`
`255`	`255`	`void NPUDeviceContext::Wait() const {`
`256`	`256`	`platform::RecordEvent record_event("NPUDeviceContext/wait");`
`257`		`- NPUDeviceGuard guard(place_.device);`
`258`	`257`	`VLOG(4) << "NPU context Wait";`
`259`		`- PADDLE_ENFORCE_NPU_SUCCESS(aclrtSynchronizeDevice());`
	`258`	`+ stream_->Wait();`
`260`	`259`	`}`
`261`	`260`
`262`	`261`	`aclrtStream NPUDeviceContext::stream() const { return stream_->raw_stream(); }`