fix typo

PaddlePaddle · Aug 9, 2021 · 2d6d013 · 2d6d013 · paddle-bot-old · Aug 9, 2021
1 parent 2a05e47
commit 2d6d013
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/paddle/fluid/memory/allocation/npu_pinned_allocator.cc b/paddle/fluid/memory/allocation/npu_pinned_allocator.cc
@@ -39,7 +39,7 @@ void NPUPinnedAllocator::ProcessEventsAndFree() {
 }
 
 Allocation *NPUPinnedAllocator::AllocateImpl(size_t size) {
-  std::lock_guard<std::mutex> lock(mutex_);
+  std::lock_guard<std::mutex> lock(mtx_);
   ProcessEventsAndFree();
   void *ptr;
   int error = posix_memalign(&ptr, kAlignment, size);
@@ -51,7 +51,7 @@ Allocation *NPUPinnedAllocator::AllocateImpl(size_t size) {
 }
 
 void NPUPinnedAllocator::FreeImpl(Allocation *allocation) {
-  std::lock_guard<std::mutex> lock(mutex_);
+  std::lock_guard<std::mutex> lock(mtx_);
   void *ptr = allocation->ptr();
   auto iter = npu_events_.find(allocation);
   aclrtEvent event = iter->second;
@@ -67,14 +67,14 @@ void NPUPinnedAllocator::FreeImpl(Allocation *allocation) {
 }
 
 uint64_t NPUPinnedAllocator::ReleaseImpl(const platform::Place &place) {
-  std::lock_guard<std::mutex> lock(mutex_);
+  std::lock_guard<std::mutex> lock(mtx_);
   // Empty implementation
   return static_cast<uint64_t>(0);
 }
 
 void NPUPinnedAllocator::RecordEvent(Allocation *allocation,
                                      aclrtStream stream) {
-  std::lock_guard<std::mutex> lock(mutex_);
+  std::lock_guard<std::mutex> lock(mtx_);
   aclrtEvent event = nullptr;
   PADDLE_ENFORCE_NPU_SUCCESS(aclrtCreateEvent(&event));
   PADDLE_ENFORCE_NPU_SUCCESS(aclrtRecordEvent(event, stream));