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 "eid_info_mgr.h"
      12              : 
      13              : #include "hccl_common.h"
      14              : #include "orion_adpt_utils.h"
      15              : 
      16              : namespace hcomm {
      17              : 
      18          116 : EidInfoMgr &EidInfoMgr::GetInstance(const uint32_t devicePhyId)
      19              : {
      20          182 :     static EidInfoMgr eidInfoMgr[MAX_MODULE_DEVICE_NUM + 1];
      21              : 
      22          116 :     uint32_t devPhyId = devicePhyId;
      23          116 :     if (devPhyId >= MAX_MODULE_DEVICE_NUM) {
      24            0 :         HCCL_WARNING("[EidInfoMgr][%s] use the backup device, devPhyId[%u] should be "
      25              :             "less than %u.", __func__, devPhyId, MAX_MODULE_DEVICE_NUM);
      26            0 :         devPhyId = MAX_MODULE_DEVICE_NUM; // 使用备份设备
      27              :     }
      28              : 
      29          116 :     eidInfoMgr[devPhyId].devPhyId_ = devPhyId;
      30          116 :     return eidInfoMgr[devPhyId];
      31              : }
      32              : 
      33           13 : HcclResult EidInfoMgr::Init()
      34              : {
      35           13 :     const RaInfo raInfo{NetworkMode::NETWORK_OFFLINE, devPhyId_};
      36           13 :     CHK_RET(RaGetDevEidInfos(raInfo, eidInfos_));
      37              : 
      38           13 :     initflag_ = true;
      39           13 :     if (eidInfos_.empty()) {
      40            0 :         HCCL_ERROR("[EidInfoMgr][%s] failed to find any eid info, "
      41              :             "devPhyId[%u].", __func__, devPhyId_);
      42            0 :         return HcclResult::HCCL_E_NOT_FOUND;
      43              :     }
      44              : 
      45           52 :     for (const auto& eidInfo : eidInfos_) {
      46           39 :         Hccl::IpAddress ipAddr{}; // 暂时使用orion ip addr 用作索引
      47           39 :         CHK_RET(CommAddrToIpAddress(eidInfo.commAddr, ipAddr));
      48           39 :         eidInfoMap_[ipAddr] = eidInfo;
      49              :     }
      50              : 
      51           13 :     return HcclResult::HCCL_SUCCESS;
      52              : }
      53              : 
      54           92 : HcclResult EidInfoMgr::GetEidInfos(std::vector<DevEidInfo> &eidInfos)
      55              : {
      56           92 :     std::unique_lock<std::mutex> lock(innerMutex_);
      57           92 :     if (!initflag_) {
      58           13 :         CHK_RET(Init());
      59              :     }
      60              : 
      61              :     // 不允许外部修改eidInfo，传递拷贝结果
      62           92 :     eidInfos.assign(eidInfos_.begin(), eidInfos_.end());
      63              : 
      64           92 :     HCCL_INFO("[EidInfoMgr][%s] found %zu eid info, devPhyId[%d].",
      65              :         __func__, eidInfos.size(), devPhyId_);
      66              : 
      67           92 :     return HCCL_SUCCESS;
      68           92 : }
      69              : 
      70           24 : HcclResult EidInfoMgr::GetEidInfoByAddr(const CommAddr &commAddr, DevEidInfo &eidInfo)
      71              : {
      72           24 :     std::unique_lock<std::mutex> lock(innerMutex_);
      73           24 :     if (!initflag_) {
      74            0 :         CHK_RET(Init());
      75              :     }
      76              : 
      77           24 :     Hccl::IpAddress ipAddr{}; // 暂时使用orion ipaddr 用作索引
      78           24 :     CHK_RET(CommAddrToIpAddress(commAddr, ipAddr));
      79              :     // 兼容上层传递ipv4查询eid
      80           24 :     const auto eidAddr = Hccl::IpAddress(ipAddr.GetEid());
      81              : 
      82           24 :     const auto addrIter = eidInfoMap_.find(eidAddr);
      83           24 :     if (addrIter == eidInfoMap_.end()) {
      84            0 :         HCCL_ERROR("[EidInfoMgr][%s] failed to find eid info by ip addr[%s], "
      85              :             "devPhyId[%u].", __func__, ipAddr.Describe().c_str(), devPhyId_);
      86            0 :         return HcclResult::HCCL_E_NOT_FOUND;
      87              :     }
      88              : 
      89           24 :     eidInfo = addrIter->second; // 触发深拷贝，避免外部修改值
      90           24 :     return HcclResult::HCCL_SUCCESS;
      91           24 : }
      92              : 
      93              : }; // namespace hcomm