Line data    Source code

       1              : /**
       2              :  * Copyright (c) 2025 Huawei Technologies Co., Ltd.
       3              :  * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
       4              :  * CANN Open Software License Agreement Version 2.0 (the "License").
       5              :  * Please refer to the License for details. You may not use this file except in compliance with the License.
       6              :  * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
       7              :  * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
       8              :  * See LICENSE in the root of the software repository for the full text of the License.
       9              :  */
      10              : 
      11              : #include "mem_mapping_manager.h"
      12              : #include "adapter_hal.h"
      13              : 
      14              : namespace hccl {
      15            0 : HcclResult MemMappingManager::MapMem(s32 deviceLogicID, void* addr, u64 size, void*& devVA)
      16              : {
      17            0 :     if (IsRequireMapping(addr, size, devVA)) {
      18              :         DevType devType;
      19            0 :         CHK_RET(hrtHalGetDeviceType(deviceLogicID, devType));
      20            0 :         drvRegisterTpye registerTpye = HOST_MEM_MAP_DEV;
      21            0 :         if ((devType == DevType::DEV_TYPE_910B) || (devType == DevType::DEV_TYPE_910_93)) {
      22              :             // 910B环境传参要特殊处理
      23            0 :             HCCL_INFO(
      24              :                 "[MemMappingManager][MapMem]aclrtHostRegister begin, addr[%p], size[%llu], flag[%u]", addr, size,
      25              :                 ACL_HOST_REGISTER_MAPPED);
      26            0 :             aclError aclRet = aclrtHostRegister(addr, size, ACL_HOST_REGISTER_MAPPED, &devVA);
      27            0 :             HCCL_INFO("[MemMappingManager][MapMem]aclrtHostRegister end, ret[%d]", aclRet);
      28            0 :             CHK_PRT_RET(
      29              :                 aclRet != ACL_SUCCESS,
      30              :                 HCCL_ERROR("[MemMappingManager][MapMem]aclrtHostRegister failed, ret[%d]", aclRet), HCCL_E_RUNTIME);
      31            0 :         } else {
      32            0 :             CHK_RET(hrtHalHostRegister(addr, size, HOST_MEM_MAP_DEV, deviceLogicID, devVA));
      33              :         }
      34            0 :         HostMappingKey hostMappingKey(reinterpret_cast<u64>(addr), size);
      35            0 :         mappedHostToDevMap_[hostMappingKey].devVA = devVA;
      36            0 :         mappedHostToDevMap_[hostMappingKey].ref.Ref();
      37            0 :         mappedHostToDevMap_[hostMappingKey].registerTpye = registerTpye;
      38              :     }
      39            0 :     return HCCL_SUCCESS;
      40              : }
      41              : 
      42              : // 先去map找内存，找到后引用计数--，减到0后做解映射，从map移除
      43            0 : HcclResult MemMappingManager::ReleaseDevVA(s32 deviceLogicID, void* addr, u64 size)
      44              : {
      45            0 :     std::unique_lock<std::mutex> lockMapping(mappedHostToDevMutex_);
      46              :     DevType devType;
      47            0 :     CHK_RET(hrtHalGetDeviceType(deviceLogicID, devType));
      48            0 :     u64 userAddr = reinterpret_cast<u64>(addr);
      49            0 :     auto iter = SearchMappingMap(userAddr, size);
      50            0 :     CHK_PRT_RET(
      51              :         (iter == mappedHostToDevMap_.end()),
      52              :         HCCL_ERROR("[MemMappingManager][ReleaseDevVA]the memory dereged isn't been reged"), HCCL_E_PARA);
      53            0 :     if (iter->second.ref.Unref() == 0) {
      54              :         // 解除内存映射，注册与解注册的 flag 保持一致
      55            0 :         if ((devType == DevType::DEV_TYPE_910B) || (devType == DevType::DEV_TYPE_910_93)) {
      56            0 :             aclError aclRet = aclrtHostUnregister(addr);
      57            0 :             CHK_PRT_RET(
      58              :                 aclRet != ACL_SUCCESS,
      59              :                 HCCL_ERROR("[MemMappingManager][ReleaseDevVA]aclrtHostUnregister failed, ret[%d]", aclRet),
      60              :                 HCCL_E_RUNTIME);
      61            0 :         } else {
      62            0 :             CHK_RET(hrtHalHostUnregisterEx(addr, deviceLogicID, iter->second.registerTpye));
      63              :         }
      64            0 :         mappedHostToDevMap_.erase(iter->first);
      65            0 :         HCCL_INFO("[MemMappingManager][ReleaseDevVA]addr[%p], size[%llu] unregister success.", addr, size);
      66              :     }
      67            0 :     return HCCL_SUCCESS;
      68            0 : }
      69              : } // namespace hccl