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 RES_TE_H
12 : #define RES_TE_H
13 :
14 : #include <map>
15 : #include <string>
16 : #include "res_pub.h"
17 :
18 : namespace Hccl {
19 :
20 2 : inline const std::map<u8, std::string>& GetTaskConciseNameMap()
21 : {
22 : static const std::map<u8, std::string> taskConciseNameMap = {
23 0 : {static_cast<u8>(TaskParamTypeVal::TASK_SDMA), "M"},
24 0 : {static_cast<u8>(TaskParamTypeVal::TASK_RDMA), "RS"},
25 0 : {static_cast<u8>(TaskParamTypeVal::TASK_SEND_PAYLOAD), "SP"},
26 0 : {static_cast<u8>(TaskParamTypeVal::TASK_REDUCE_INLINE), "IR"},
27 0 : {static_cast<u8>(TaskParamTypeVal::TASK_UB_REDUCE_INLINE), "IR"},
28 0 : {static_cast<u8>(TaskParamTypeVal::TASK_REDUCE_TBE), "R"},
29 0 : {static_cast<u8>(TaskParamTypeVal::TASK_UB), "WorR"},
30 0 : {static_cast<u8>(TaskParamTypeVal::TASK_NOTIFY_RECORD), "NR"},
31 0 : {static_cast<u8>(TaskParamTypeVal::TASK_NOTIFY_WAIT), "NW"},
32 0 : {static_cast<u8>(TaskParamTypeVal::TASK_SEND_NOTIFY), "SN"},
33 0 : {static_cast<u8>(TaskParamTypeVal::TASK_WRITE_WITH_NOTIFY), "WN"},
34 0 : {static_cast<u8>(TaskParamTypeVal::TASK_UB_INLINE_WRITE), "IW"},
35 0 : {static_cast<u8>(TaskParamTypeVal::TASK_WRITE_REDUCE_WITH_NOTIFY), "WRN"},
36 0 : {static_cast<u8>(TaskParamTypeVal::TASK_CCU), "CCU"},
37 0 : {static_cast<u8>(TaskParamTypeVal::TASK_AICPU_KERNEL), "AIK"},
38 19 : };
39 2 : return taskConciseNameMap;
40 1 : }
41 :
42 : } // namespace Hccl
43 :
44 : #endif
|