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 CCU_RES_SPECIFICATIONS_H
12 : #define CCU_RES_SPECIFICATIONS_H
13 :
14 : #include "ccu_device_manager.h"
15 :
16 : namespace Hccl {
17 :
18 : constexpr uint32_t CCU_RESOURCE_SIZE = 72 * 1024 * 1024; // CCU资源空间大小
19 : constexpr uint64_t CCU_V1_CCUM_OFFSET = 0x800000; // V1 CCUM 偏移，位于CCUA之后
20 :
21 : constexpr uint64_t CCU_V1_WQE_BASIC_BLOCK_OFFSET = (CCU_V1_CCUM_OFFSET + 0x800000);
22 :
23 : constexpr uint32_t CCU_ONE_WQE_SIZE = 64; // Bytes
24 : constexpr uint32_t CCU_WQE_NUM_PER_SQE = 4; // URMA 约束每个SQE包含4个WQEBB
25 : constexpr uint32_t CCU_MIN_SQ_DEPTH = 16;
26 : constexpr uint32_t CCU_MAX_SQ_DEPTH = 256;
27 : constexpr uint16_t CCU_START_TA_JETTY_ID = 1024; // IMP给系统预留给CCU的jetty Id起始编号
28 : constexpr uint32_t CCU_SQ_BUFFER_SIZE = 256 * 1024; // ccu 每个jetty sq buffer size 固定为256k
29 : constexpr uint32_t CCU_WQEBB_RESOURCE_NUM = 4096;
30 : constexpr uint32_t CCU_V1_PER_DIE_PFE_RESERVED_NUM = 16; // ccu 每个IO die预留16个PFE表
31 : constexpr uint32_t CCU_PER_DIE_JETTY_RESERVED_NUM = 128; // ccu 每个IO die默认jetty数量
32 : constexpr uint32_t CCU_MEM_INFO_SIZE = 64;
33 :
34 : constexpr uint64_t CCU_RESOURCE_INS_RESERVE_SIZE = 0x100000; // INS预留空间1M
35 : constexpr uint64_t CCU_V1_RESOURCE_GSA_RESERVE_SIZE = 0x8000; // v1 GSA预留空间32K
36 : constexpr uint16_t CCU_RESOURCE_XN_PER_SIZE = 8;
37 :
38 : constexpr uint32_t MOVE_16_BITS = 16;
39 : constexpr uint32_t MOVE_24_BITS = 24;
40 :
41 : // CcuBlockResStrategy 定义了资源管理块资源类型的块大小
42 : struct CcuBlockResStrategy {
43 : uint32_t loopNum{16};
44 : uint32_t ckeNum{16};
45 : uint32_t msNum{128};
46 : uint32_t missionNum{2};
47 : };
48 :
49 : enum CcuMemTypeBitmap : uint64_t;
50 :
51 : // 当前需要申请权限的内存为18块
52 46 : inline const std::array<CcuMemTypeBitmap, 18>& GetMemTypeVector() {
53 : static const std::array<CcuMemTypeBitmap, 18> kMemTypeArray = {{
54 : CCU_MEMTYPE_INS,
55 : CCU_MEMTYPE_GSA,
56 : CCU_MEMTYPE_XN,
57 : CCU_MEMTYPE_CKE,
58 : CCU_MEMTYPE_PFE,
59 : CCU_MEMTYPE_CHN,
60 : CCU_MEMTYPE_JETTY_CTX,
61 : CCU_MEMTYPE_MISSION_CTX,
62 : CCU_MEMTYPE_LOOP_CTX,
63 : CCU_MEMTYPE_MISSION_SQE,
64 : CCU_MEMTYPE_CQE_BLOCK0,
65 : CCU_MEMTYPE_CQE_BLOCK1,
66 : CCU_MEMTYPE_CQE_BLOCK2,
67 : CCU_MEMTYPE_WQEBB,
68 : CCU_MEMTYPE_MS_BLOCK0,
69 : CCU_MEMTYPE_MS_BLOCK1,
70 : CCU_MEMTYPE_MS_BLOCK2,
71 : CCU_MEMTYPE_MS_BLOCK3
72 : }};
73 46 : return kMemTypeArray;
74 : }
75 :
76 11 : inline uint64_t GetCombinedMemTypeBitmap() {
77 11 : const auto& memTypes = GetMemTypeVector();
78 11 : uint64_t combined = 0;
79 :
80 209 : for (const auto& memType : memTypes) {
81 198 : combined |= static_cast<uint64_t>(memType);
82 : }
83 :
84 11 : return combined;
85 : }
86 :
87 : struct CcuMemInfo;
88 :
89 : struct CcuResSpecInfo {
90 : // 基础信息
91 : uint32_t msId{0};
92 : uint32_t missionKey{0};
93 : uint64_t resourceAddr{0};
94 : // 通过能力寄存器获取
95 : uint32_t loopEngineNum{0};
96 : uint32_t missionNum{0};
97 : uint32_t instructionNum{0};
98 : uint32_t xnNum{0};
99 : uint32_t gsaNum{0};
100 : uint32_t msNum{0};
101 : uint32_t ckeNum{0};
102 : uint32_t jettyNum{0};
103 : uint32_t channelNum{0};
104 : uint32_t pfeNum{0};
105 : // 额外资源信息
106 : uint32_t wqeBBNum{CCU_WQEBB_RESOURCE_NUM};
107 : uint32_t dieNum{MAX_CCU_IODIE_NUM};
108 :
109 : // 内存信息
110 : uint32_t memNum{0};
111 : std::array<CcuMemInfo, CCU_MEM_INFO_SIZE> memInfoList{};
112 : };
113 :
114 : class CcuResSpecifications {
115 : public:
116 : CcuResSpecifications(const CcuResSpecifications &that) = delete;
117 : CcuResSpecifications &operator=(const CcuResSpecifications &that) = delete;
118 : ~CcuResSpecifications() = default;
119 :
120 : void Deinit();
121 : void Init();
122 :
123 : static CcuResSpecifications& GetInstance(const int32_t deviceLogicId);
124 :
125 : CcuVersion GetCcuVersion() const;
126 : bool GetAXFlag() const;
127 : HcclResult GetDieEnableFlag(const uint8_t dieId, bool &dieEnableFlag) const;
128 :
129 : HcclResult GetResourceAddr(const uint8_t dieId, uint64_t &resourceAddr) const;
130 : HcclResult GetXnBaseAddr(const uint8_t dieId, uint64_t &xnBaseAddr) const;
131 :
132 : HcclResult GetMsId(const uint8_t dieId, uint32_t &msId) const;
133 : HcclResult GetMissionKey(const uint8_t dieId, uint32_t &missionKey) const;
134 :
135 : // 寄存器资源
136 : HcclResult GetMissionNum(const uint8_t dieId, uint32_t &missionNum) const;
137 : HcclResult GetMsNum(const uint8_t dieId, uint32_t &msNum) const;
138 : HcclResult GetLoopEngineNum(const uint8_t dieId, uint32_t &loopNum) const;
139 : HcclResult GetCkeNum(const uint8_t dieId, uint32_t &ckeNum) const;
140 : HcclResult GetXnNum(const uint8_t dieId, uint32_t &xnNum) const;
141 : HcclResult GetInstructionNum(const uint8_t dieId, uint32_t &instrNum) const;
142 : HcclResult GetGsaNum(const uint8_t dieId, uint32_t &gsaNum) const;
143 :
144 : // channel资源
145 : HcclResult GetChannelNum(const uint8_t dieId, uint32_t &channelNum) const;
146 : HcclResult GetJettyNum(const uint8_t dieId, uint32_t &jettyNum) const;
147 : HcclResult GetPfeReservedNum(const uint8_t dieId, uint32_t &pfeNum) const;
148 : HcclResult GetPfeNum(const uint8_t dieId, uint32_t &pfeNum) const;
149 : HcclResult GetWqeBBNum(const uint8_t dieId, uint32_t &wqeBBNum) const;
150 :
151 : // ccu mem info
152 : HcclResult GetCcuMemInfoList(const uint8_t dieId, struct CcuMemInfo *memInfoList, uint32_t &count);
153 :
154 : private:
155 : int32_t devLogicId{0};
156 : uint32_t devPhyId{0};
157 : std::array<bool, MAX_CCU_IODIE_NUM> dieEnableFlags{}; // 根据资源规格的记录可用的die
158 : std::array<CcuResSpecInfo, MAX_CCU_IODIE_NUM> resSpecs{};
159 : bool ifInit{false};
160 : bool isAX{false};
161 : CcuVersion ccuVersion{CcuVersion::CCU_INVALID};
162 :
163 : explicit CcuResSpecifications() = default;
164 :
165 : HcclResult Init_();
166 : };
167 :
168 : HcclResult CheckDieValid(const std::string &funcName, const int32_t devLogicId, const uint8_t dieId,
169 : bool dieEnableFlag);
170 :
171 : using GetResSpecFunc = HcclResult (CcuResSpecifications::*)(const uint8_t, uint32_t&) const;
172 : using ResSpecFuncPair = std::pair<ResType, GetResSpecFunc>;
173 : constexpr ResSpecFuncPair GET_RES_SPEC_FUNC_ARRAY[] = {
174 : {ResType::LOOP, &CcuResSpecifications::GetLoopEngineNum},
175 : {ResType::MS, &CcuResSpecifications::GetMsNum},
176 : {ResType::CKE, &CcuResSpecifications::GetCkeNum},
177 : {ResType::XN, &CcuResSpecifications::GetXnNum},
178 : {ResType::GSA, &CcuResSpecifications::GetGsaNum},
179 : {ResType::INS, &CcuResSpecifications::GetInstructionNum},
180 : {ResType::MISSION, &CcuResSpecifications::GetMissionNum}
181 : };
182 :
183 : }; // namespace Hccl
184 : #endif // CCU_RES_SPECIFICATIONS_H
|
