change malloc physical

Author: youxiao

mooncake-store/src/utils.cpp

@@ -79,13 +79,31 @@ void *allocate_buffer_allocator_memory(size_t total_size,
     }
 #ifdef USE_ASCEND_DIRECT
     if (protocol == "ascend" && total_size > 0) {
-        void *buffer = nullptr;
-        auto ret = aclrtMallocHost(&buffer, total_size);
+        aclrtDrvMemHandle handle = nullptr;
+        aclrtPhysicalMemProp prop = {};
+        prop.handleType = ACL_MEM_HANDLE_TYPE_NONE;
+        prop.allocationType = ACL_MEM_ALLOCATION_TYPE_PINNED;
+        prop.memAttr = ACL_DDR_MEM_HUGE;
+        prop.location.type = ACL_MEM_LOCATION_TYPE_HOST;
+        prop.location.id = 0;
+        prop.reserve = 0;
+        auto ret = aclrtMallocPhysical(&handle, total_size, &prop, 0);
         if (ret != ACL_ERROR_NONE) {
             LOG(ERROR) << "Failed to allocate memory: " << ret;
             return nullptr;
         }
-        return buffer;
+        void *va;
+        ret = aclrtReserveMemAddress(&va, total_size, 0, nullptr, 1);
+        if (ret != ACL_ERROR_NONE) {
+            LOG(ERROR) << "Failed to reserve memory: " << ret;
+            return nullptr;
+        }
+        ret = aclrtMapMem(va, total_size, 0, handle, 0);
+        if (ret != ACL_ERROR_NONE) {
+            LOG(ERROR) << "Failed to map memory: " << ret;
+            return nullptr;
+        }
+        return va;
     }
 #endif
     // Allocate aligned memory
@@ -95,7 +113,9 @@ void *allocate_buffer_allocator_memory(size_t total_size,
 void free_memory(const std::string &protocol, void *ptr) {
 #ifdef USE_ASCEND_DIRECT
     if (protocol == "ascend") {
-        aclrtFreeHost(ptr);
+        aclrtUnmapMem(ptr);
+        aclrtReleaseMemAddress(ptr);
+        // TODO retain & free physical
         return;
     }
 #endif