fix compile problem (#31850)

PaddlePaddle · Mar 24, 2021 · a934888 · a934888 · paddle-bot-old · Mar 24, 2021
1 parent 0279486
commit a934888
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/paddle/fluid/operators/npu_op_runner.cc b/paddle/fluid/operators/npu_op_runner.cc
@@ -65,9 +65,10 @@ aclFormat ConvertToNpuFormat(DataLayout layout) {
 }
 
 aclrtStream GetCurrentNPUStream() {
-  int device_id = GetCurrentNPUDeviceId();
+  int device_id = platform::GetCurrentNPUDeviceId();
   platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance();
-  auto &dev_ctx = *pool.Get(platform::NPUPlace(device_id));
+  auto *dev_ctx = static_cast<platform::NPUDeviceContext *>(
+      pool.Get(platform::NPUPlace(device_id)));
   return dev_ctx->stream();
 }
 

diff --git a/paddle/fluid/operators/npu_op_runner.h b/paddle/fluid/operators/npu_op_runner.h
@@ -69,7 +69,7 @@ class NpuOpRunner {
 
   std::vector<aclDataBuffer *> &GetOutputBuffers();
 
-  void Run(aclrtStream stream == nullptrr);
+  void Run(aclrtStream stream = nullptr);
 
  private:
   aclTensorDesc *CreateTensorDesc(Tensor tensor);