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_CCU_INSTRUCTION_ALL_REDUCE_MESH_1D_MEM2MEM_H_
12 : #define HCCLV2_CCU_INSTRUCTION_ALL_REDUCE_MESH_1D_MEM2MEM_H_
13 :
14 : #include "template_utils.h"
15 : #include "instruction.h"
16 : #include "ins_queue.h"
17 : #include "ccu_context_utils.h"
18 : #include "ccu_ctx_signature.h"
19 : #include "ccu_ins.h"
20 : #include "ccu_rank_group.h"
21 :
22 : namespace Hccl {
23 :
24 : // 为AllReduceMesh1D实现的CCUIns、CCUCtxArg与CCUTaskArg
25 : class CcuCtxArgAllReduceMeshMem2Mem1D : public CcuCtxArg {
26 : public:
27 0 : explicit CcuCtxArgAllReduceMeshMem2Mem1D(const std::vector<uint64_t> &dimSize, uint32_t rankId,
28 : const CollAlgOperator &op,
29 : const std::vector<std::vector<RankId>> &tempVTopo)
30 0 : : dimSize_(dimSize), rankId_(rankId), op_(op), tempVTopo_(tempVTopo)
31 : {
32 0 : }
33 0 : CcuCtxSignature GetCtxSignature() const override
34 : {
35 0 : CcuCtxSignature signature;
36 0 : GenerateCcuCtxSignature(signature, CcuInstType::CCU_ALL_REDUCE_MESH_1D_MEM2MEM, op_,
37 0 : tempVTopo_);
38 0 : return signature;
39 0 : }
40 : std::vector<uint64_t> dimSize_;
41 : uint32_t rankId_;
42 : CollAlgOperator op_;
43 : std::vector<std::vector<RankId>> tempVTopo_;
44 : };
45 :
46 : class CcuTaskArgAllReduceMeshMem2Mem1D : public CcuTaskArg {
47 : public:
48 0 : explicit CcuTaskArgAllReduceMeshMem2Mem1D(uint64_t inputAddr, uint64_t outputAddr, uint64_t token,
49 : uint64_t scratchAddr, uint64_t inputSliceStride,
50 : uint64_t outputSliceStride, uint64_t inputRepeatStride,
51 : uint64_t outputRepeatStride, uint64_t normalSliceSize,
52 : uint64_t lastSliceSize, uint64_t mySliceSize,
53 : uint64_t isInputOutputEqual)
54 0 : : inputAddr_(inputAddr), outputAddr_(outputAddr), token_(token), scratchAddr_(scratchAddr),
55 0 : inputSliceStride_(inputSliceStride), outputSliceStride_(outputSliceStride),
56 0 : inputRepeatStride_(inputRepeatStride), outputRepeatStride_(outputRepeatStride),
57 0 : normalSliceSize_(normalSliceSize), lastSliceSize_(lastSliceSize), mySliceSize_(mySliceSize),
58 0 : isInputOutputEqual_(isInputOutputEqual)
59 : {
60 0 : }
61 :
62 : uint64_t inputAddr_;
63 : uint64_t outputAddr_;
64 : uint64_t token_;
65 : uint64_t scratchAddr_;
66 : uint64_t inputSliceStride_;
67 : uint64_t outputSliceStride_;
68 : uint64_t inputRepeatStride_;
69 : uint64_t outputRepeatStride_;
70 :
71 : uint64_t normalSliceSize_;
72 : uint64_t lastSliceSize_;
73 : uint64_t mySliceSize_;
74 : uint64_t isInputOutputEqual_;
75 : };
76 :
77 : class CcuInstructionAllReduceMeshMem2Mem1D : public CcuInstruction {
78 : public:
79 0 : CcuInstructionAllReduceMeshMem2Mem1D() : CcuInstruction()
80 : {
81 0 : }
82 :
83 0 : void Init(uint32_t rankId, const CollAlgOperator &op, const std::vector<std::vector<RankId>> &tempVTopo,
84 : uint64_t inputAddr, uint64_t outputAddr, uint64_t token, uint64_t scratchAddr, uint64_t inputSliceStride,
85 : uint64_t outputSliceStride, uint64_t inputRepeatStride, uint64_t outputRepeatStride,
86 : uint64_t normalSliceSize, uint64_t lastSliceSize, uint64_t mySliceSize, uint64_t isInputOutputEqual)
87 : {
88 0 : u32 maxDimNum = 1;
89 0 : if (tempVTopo.size() != maxDimNum) {
90 0 : THROW<InvalidParamsException>(StringFormat(
91 : "[CcuInstructionAllReduceMesh1D] tempVTopo size is not 1, size is [%zu].", tempVTopo.size()));
92 : }
93 0 : dimSize_.push_back(tempVTopo[0].size());
94 0 : rankId_ = rankId;
95 0 : op_ = op;
96 0 : tempVTopo_ = tempVTopo;
97 0 : inputAddr_ = inputAddr;
98 0 : outputAddr_ = outputAddr;
99 0 : scratchAddr_ = scratchAddr;
100 0 : inputSliceStride_ = inputSliceStride;
101 0 : outputSliceStride_ = outputSliceStride;
102 0 : inputRepeatStride_ = inputRepeatStride;
103 0 : outputRepeatStride_ = outputRepeatStride;
104 0 : normalSliceSize_ = normalSliceSize;
105 0 : lastSliceSize_ = lastSliceSize;
106 0 : mySliceSize_ = mySliceSize;
107 0 : token_ = token;
108 0 : isInputOutputEqual_ = isInputOutputEqual;
109 0 : return;
110 : }
111 :
112 0 : CcuInstType GetInstType() const override
113 : {
114 0 : HCCL_INFO("CcuInstructionAllReduceMesh1D instype is CCU_ALL_REDUCE_MESH_1D_MEM2MEM.");
115 0 : return instType_;
116 : }
117 :
118 0 : std::string Describe() const override
119 : {
120 0 : return StringFormat("CcuInstructionAllReduceMesh1D rankId [%u], instType[%s]", rankId_,
121 0 : instType_.Describe().c_str());
122 : }
123 :
124 0 : std::unique_ptr<CcuCtxArg> GetCtxArg() const override
125 : {
126 0 : return std::make_unique<CcuCtxArgAllReduceMeshMem2Mem1D>(dimSize_, rankId_, op_, tempVTopo_);
127 : }
128 :
129 : void SetInstType(CcuInstType instType)
130 : {
131 : instType_ = instType;
132 : }
133 :
134 0 : std::unique_ptr<CcuTaskArg> GetTaskArg() const override
135 : {
136 0 : return std::make_unique<CcuTaskArgAllReduceMeshMem2Mem1D>(
137 0 : inputAddr_, outputAddr_, token_, scratchAddr_, inputSliceStride_, outputSliceStride_, inputRepeatStride_,
138 0 : outputRepeatStride_, normalSliceSize_, lastSliceSize_, mySliceSize_, isInputOutputEqual_);
139 : }
140 :
141 : private:
142 : CcuInstType instType_ = CcuInstType::CCU_ALL_REDUCE_MESH_1D_MEM2MEM;
143 : std::vector<uint64_t> dimSize_;
144 : uint32_t rankId_{0};
145 : CollAlgOperator op_;
146 : std::vector<std::vector<RankId>> tempVTopo_;
147 : uint64_t inputAddr_{0};
148 : uint64_t outputAddr_{0};
149 : uint64_t token_{0};
150 : uint64_t scratchAddr_{0};
151 : uint64_t inputSliceStride_{0};
152 : uint64_t outputSliceStride_{0};
153 : uint64_t inputRepeatStride_{0};
154 : uint64_t outputRepeatStride_{0};
155 : uint64_t normalSliceSize_{0};
156 : uint64_t lastSliceSize_{0};
157 : uint64_t mySliceSize_{0};
158 : uint64_t isInputOutputEqual_{0};
159 : };
160 :
161 : } // namespace Hccl
162 : #endif // HCCLV2_CCU_INSTRUCTION_ALL_REDUCE_MESH_1D_MEM2MEM_H_
|
