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