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 : #include "ccu_rep.h"
12 : #include "string_util.h"
13 :
14 : namespace Hccl {
15 : namespace CcuRep {
16 :
17 1 : CcuRepStore::CcuRepStore(const Variable& var, uint64_t addr) : var(var), addr(addr)
18 : {
19 1 : type = CcuRepType::STORE;
20 1 : instrCount = 7; // 7: Store包含7条指令
21 1 : }
22 :
23 1 : bool CcuRepStore::Translate(CcuInstr*& instr, uint16_t& instrId, const TransDep& dep)
24 : {
25 1 : this->instrId = instrId;
26 1 : translated = true;
27 1 : uint64_t varAddr = dep.xnBaseAddr + CCU_RESOURCE_XN_PER_SIZE * var.Id();
28 :
29 1 : LoadImdToGSAInstr(instr++, dep.commGsa[0], addr);
30 1 : LoadImdToGSAInstr(instr++, dep.commGsa[1], varAddr);
31 1 : LoadImdToXnInstr(instr++, dep.commXn[0], dep.memTokenInfo, CCU_LOAD_TO_XN_SEC_INFO);
32 1 : LoadImdToXnInstr(instr++, dep.commXn[1], dep.ccuResSpaceTokenInfo, CCU_LOAD_TO_XN_SEC_INFO);
33 1 : LoadImdToXnInstr(instr++, dep.commXn[2], CCU_RESOURCE_XN_PER_SIZE);
34 1 : TransLocMemToLocMemInstr(
35 1 : instr++, dep.commGsa[0], dep.commXn[0], dep.commGsa[1], dep.commXn[1], dep.commXn[2],
36 1 : dep.reserveChannalId[0], dep.commSignal, mask, 0, 0, 1, 1);
37 1 : SetCKEInstr(instr++, 0, 0, dep.commSignal, mask, 1);
38 1 : instrId += instrCount;
39 :
40 1 : return translated;
41 : }
42 :
43 1 : std::string CcuRepStore::Describe() { return StringFormat("Store([%u], [%llu])", var.Id(), addr); }
44 :
45 : }; // namespace CcuRep
46 : }; // namespace Hccl
|