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              : #ifndef HCCLV2_TYPES_H
      12              : #define HCCLV2_TYPES_H
      13              : 
      14              : #include <cstdint>
      15              : #include <unordered_map>
      16              : #include "hccl/base.h"
      17              : #include "enum_factory_legacy.h"
      18              : 
      19              : namespace Hccl {
      20              : using char_t = char;
      21              : 
      22              : using RankId = s32;
      23              : using QId = u32;
      24              : using DevId = u32;
      25              : using Timeout = u32;
      26              : 
      27              : using SockPort = u16;
      28              : 
      29              : // 将用户配置的HcclCommConfig.hcclOpExpansionMode -- int32_t 转为相应枚举
      30         2198 : MAKE_ENUM(HcclAccelerator, DEFAULT, HOSTCPU_TS, AICPU_TS, AIV, AIV_ONLY, CCU_MS, CCU_SCHED, AICPU);
      31              : 
      32              : // 通信域粒度加速模式
      33       263306 : MAKE_ENUM(AcceleratorState, CCU_MS, CCU_SCHED, CCU_FALLBACK, AIV, AIV_ONLY, AICPU_TS, HOSTCPU_TS, AICPU);
      34              : 
      35              : // AcceleratorState是通信域内部加速模式
      36              : struct OpExecuteConfig {
      37              :     AcceleratorState accState = AcceleratorState::CCU_MS;
      38              : };
      39              : 
      40              : struct AcceleratorEnumHash {
      41              :     template <typename T>
      42       134760 :     std::size_t operator()(T t) const
      43              :     {
      44       134760 :         return static_cast<std::size_t>(t);
      45              :     }
      46              : };
      47              : 
      48              : const std::unordered_map<AcceleratorState, std::string, AcceleratorEnumHash> AcceleratorStateToString
      49              :     = {{AcceleratorState::CCU_MS, "CCU_MS"},
      50              :        {AcceleratorState::CCU_SCHED, "CCU_SCHED"},
      51              :        {AcceleratorState::CCU_FALLBACK, "CCU_FALLBACK"},
      52              :        {AcceleratorState::AIV, "AIV"},
      53              :        {AcceleratorState::AIV_ONLY, "AIV_ONLY"},
      54              :        {AcceleratorState::AICPU_TS, "AICPU_TS"},
      55              :        {AcceleratorState::HOSTCPU_TS, "HOSTCPU_TS"},
      56              :        {AcceleratorState::AICPU, "AICPU"}};
      57              : 
      58          520 : MAKE_ENUM(
      59              :     HcclOpType, HCCL_INVALID, HCCL_BROADCAST, HCCL_ALLREDUCE, HCCL_REDUCE, HCCL_SEND, HCCL_RECEIVE, HCCL_ALLGATHER,
      60              :     HCCL_REDUCE_SCATTER, HCCL_ALLTOALLV, HCCL_ALLTOALLVC, HCCL_ALLTOALL, HCCL_GATHER, HCCL_SCATTER,
      61              :     HCCL_BATCH_SEND_RECV, HCCL_BATCH_PUT, HCCL_BATCH_GET, HCCL_ALLGATHER_V, HCCL_REDUCE_SCATTER_V, HCCL_BATCH_WRITE,
      62              :     HCCL_HALF_ALLTOALLV, HCCL_ALL, HCCL_FINALIZE = 100, HCCL_INTER_GROUP_SYNC, HCCL_INIT, HCCL_BARRIER, HCCL_MAX);
      63              : 
      64              : } // namespace Hccl
      65              : 
      66              : #endif // HCCLV2_TYPES_H