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("[MemMappingManager][MapMem]aclrtHostRegister begin, addr[%p], size[%llu], flag[%u]",
      24              :                 addr, size, ACL_HOST_REGISTER_MAPPED);
      25            0 :             aclError aclRet = aclrtHostRegister(addr, size, ACL_HOST_REGISTER_MAPPED, &devVA);
      26            0 :             HCCL_INFO("[MemMappingManager][MapMem]aclrtHostRegister end, ret[%d]", aclRet);
      27            0 :             CHK_PRT_RET(aclRet != ACL_SUCCESS,
      28              :                 HCCL_ERROR("[MemMappingManager][MapMem]aclrtHostRegister failed, ret[%d]", aclRet), HCCL_E_RUNTIME);
      29            0 :         } else {
      30            0 :             CHK_RET(hrtHalHostRegister(addr, size, HOST_MEM_MAP_DEV, deviceLogicID, devVA));
      31              :         }
      32            0 :         HostMappingKey hostMappingKey(reinterpret_cast<u64>(addr), size);
      33            0 :         mappedHostToDevMap_[hostMappingKey].devVA = devVA;
      34            0 :         mappedHostToDevMap_[hostMappingKey].ref.Ref();
      35            0 :         mappedHostToDevMap_[hostMappingKey].registerTpye = registerTpye;
      36              :     }
      37            0 :     return HCCL_SUCCESS;
      38              : }
      39              : 
      40              : // 先去map找内存，找到后引用计数--，减到0后做解映射，从map移除
      41            0 : HcclResult MemMappingManager::ReleaseDevVA(s32 deviceLogicID, void *addr, u64 size)
      42              : {
      43            0 :     std::unique_lock<std::mutex> lockMapping(mappedHostToDevMutex_);
      44              :     DevType devType;
      45            0 :     CHK_RET(hrtHalGetDeviceType(deviceLogicID, devType));
      46            0 :     u64 userAddr = reinterpret_cast<u64>(addr);
      47            0 :     auto iter = SearchMappingMap(userAddr, size);
      48            0 :     CHK_PRT_RET((iter == mappedHostToDevMap_.end()),
      49              :         HCCL_ERROR("[MemMappingManager][ReleaseDevVA]the memory dereged isn't been reged"), HCCL_E_PARA);
      50            0 :     if (iter->second.ref.Unref() == 0) {
      51              :         // 解除内存映射，注册与解注册的 flag 保持一致
      52            0 :         if ((devType == DevType::DEV_TYPE_910B) || (devType == DevType::DEV_TYPE_910_93)) {
      53            0 :             aclError aclRet = aclrtHostUnregister(addr);
      54            0 :             CHK_PRT_RET(aclRet != ACL_SUCCESS,
      55              :                 HCCL_ERROR("[MemMappingManager][ReleaseDevVA]aclrtHostUnregister failed, ret[%d]", aclRet),
      56              :                 HCCL_E_RUNTIME);
      57            0 :         } else {
      58            0 :             CHK_RET(hrtHalHostUnregisterEx(addr, deviceLogicID, iter->second.registerTpye));
      59              :         }
      60            0 :         mappedHostToDevMap_.erase(iter->first);
      61            0 :         HCCL_INFO("[MemMappingManager][ReleaseDevVA]addr[%p], size[%llu] unregister success.", addr, size);
      62              :     }
      63            0 :     return HCCL_SUCCESS;
      64            0 : }
      65              : }