From f75074db90281db72bcc11a0bde53144f457b200 Mon Sep 17 00:00:00 2001
From: chuxing <chuxing@huawei.com>
Date: Thu, 12 Nov 2020 15:43:42 +0800
Subject: [PATCH] fix padding

---
 ge/hybrid/common/npu_memory_allocator.cc | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/ge/hybrid/common/npu_memory_allocator.cc b/ge/hybrid/common/npu_memory_allocator.cc
index d3ebb347..b8a137cb 100644
--- a/ge/hybrid/common/npu_memory_allocator.cc
+++ b/ge/hybrid/common/npu_memory_allocator.cc
@@ -69,11 +69,10 @@ void *NpuMemoryAllocator::Allocate(std::size_t size, AllocationAttr *attr) {
       if (attr->padding_ > 0) {
         padding = attr->padding_;
       }
-
-      // padding up to multiple of padding, and add extra padding
-      allocate_size = (size + 2 * padding - 1) / padding * padding;
-      GELOGD("Padding size %ld by %d. final size = %zu.", size, padding, allocate_size);
     }
+    // padding up to multiple of padding, and add extra padding
+    allocate_size = (size + 2 * padding - 1) / padding * padding;
+    GELOGD("Padding size %ld by %d. final size = %zu.", size, padding, allocate_size);
     buffer = MemManager::Instance()
                  .CachingInstance(RT_MEMORY_HBM)
                  .Malloc(allocate_size, reinterpret_cast<uint8_t *>(try_reuse_addr), device_id_);