From 41b4fbaff086a5e75f54fcac28d494908a0dd2e1 Mon Sep 17 00:00:00 2001
From: ascend-direct-dev <lbjyx@outlook.com>
Date: Thu, 19 Mar 2026 15:23:52 +0800
Subject: [PATCH] ascend fabric mem support failover to 2M malloc

---
 .../src/transport/ascend_transport/ascend_allocator.cpp  | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/mooncake-transfer-engine/src/transport/ascend_transport/ascend_allocator.cpp b/mooncake-transfer-engine/src/transport/ascend_transport/ascend_allocator.cpp
index 4ca2210f..31afbdcb 100644
--- a/mooncake-transfer-engine/src/transport/ascend_transport/ascend_allocator.cpp
+++ b/mooncake-transfer-engine/src/transport/ascend_transport/ascend_allocator.cpp
@@ -48,8 +48,13 @@ int allocate_physical_memory(size_t total_size, aclrtDrvMemHandle &handle) {
         prop.location.id = 0;
         ret = aclrtMallocPhysical(&handle, total_size, &prop, 0);
         if (ret != ACL_ERROR_NONE) {
-            LOG(ERROR) << "Failed to allocate memory: " << ret;
-            return -1;
+            LOG(INFO) << "Malloc failed, try smaller page instead.";
+            prop.memAttr = ACL_MEM_P2P_HUGE;
+            ret = aclrtMallocPhysical(&handle, total_size, &prop, 0);
+            if (ret != ACL_ERROR_NONE) {
+                LOG(ERROR) << "Failed to allocate memory: " << ret;
+                return -1;
+            }
         }
     }
     return 0;