* Copyright (c) 2025 Huawei Technologies Co., Ltd.
* This program is free software, you can redistribute it and/or modify it under the terms and conditions of
* CANN Open Software License Agreement Version 2.0 (the "License").
* Please refer to the License for details. You may not use this file except in compliance with the License.
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
* INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
* See LICENSE in the root of the software repository for the full text of the License.
*/
#ifndef INC_GRAPH_UTILS_OP_DESC_UTILS_H_
#define INC_GRAPH_UTILS_OP_DESC_UTILS_H_
#include <string>
#include <vector>
#include "graph/def_types.h"
#include "graph/gnode.h"
#include "graph/node.h"
#include "graph/operator.h"
#include "graph/runtime_inference_context.h"
namespace ge {
using ConstGeTensorBarePtr = const GeTensor *;
class OpDescUtils {
public:
template <class T>
using Vistor = RangeVistor<T, std::shared_ptr<OpDesc>>;
using GetConstInputOnRuntimeFun =
std::function<ge::graphStatus(const ConstNodePtr &node, const size_t index, ge::GeTensorPtr &tensor)>;
OpDescUtils() = default;
~OpDescUtils() = default;
static bool HasQuantizeFactorParams(const OpDescPtr& op_desc);
static bool HasQuantizeFactorParams(const OpDesc& op_desc);
static std::vector<ge::NodePtr> GetConstInputNode(const ge::Node& node);
static std::vector<NodeToOutAnchor> GetConstInputNodeAndAnchor(const ge::Node &node);
static std::vector<ConstGeTensorPtr> GetInputData(const std::vector<ge::NodePtr>& input_nodes);
static std::vector<ConstGeTensorPtr> GetWeightsFromNodes(
const std::vector<NodeToOutAnchor>& input_nodes_2_out_anchors);
static std::vector<ConstGeTensorPtr> GetWeights(const ge::Node& node);
static std::vector<ConstGeTensorPtr> GetWeights(const ge::ConstNodePtr& node);
static std::vector<GeTensorPtr> MutableWeights(const ge::Node& node);
static std::vector<GeTensorPtr> MutableWeights(const ge::NodePtr node);
static graphStatus SetWeights(ge::Node& node, const std::vector<ge::GeTensorPtr>& weights);
static graphStatus SetWeights(ge::NodePtr node, const std::vector<ge::GeTensorPtr> &weights);
static graphStatus SetWeights(ge::Node &node, const std::map<int, ge::GeTensorPtr> &weights_map);
static graphStatus ClearWeights(const ge::NodePtr node);
static graphStatus SetNoneConstNodeWeights(ge::Node &node, const std::map<int, ge::GeTensorPtr> &weights_map);
static graphStatus SetNoneConstNodeWeights(ge::Node &node, const std::vector<ge::GeTensorPtr> &weights);
static bool ClearInputDesc(const ge::OpDescPtr op_desc, const uint32_t index);
static bool ClearInputDesc(const ge::NodePtr& node);
static bool ClearOutputDesc(const ge::OpDescPtr& op_desc, const uint32_t index);
static bool ClearOutputDesc(const ge::NodePtr& node);
static std::vector<ge::NodePtr> GetConstInputs(const ge::Node& node, const uint32_t depth = 64U);
static std::vector<ge::NodePtr> GetConstInputs(const ge::ConstNodePtr& node);
static size_t GetNonConstInputsSize(const ge::Node& node);
static size_t GetNonConstInputsSize(const ge::ConstNodePtr node);
static GeTensorDesc GetNonConstInputTensorDesc(const ge::Node& node, const size_t index_non_const = 0UL);
static GeTensorDesc GetNonConstInputTensorDesc(const ge::ConstNodePtr& node, const size_t index_non_const = 0UL);
static bool GetNonConstInputIndex(const ge::Node& node, const size_t index_non_const, size_t& index);
static bool GetNonConstInputIndex(const ge::ConstNodePtr& node, const size_t index_non_const, size_t& index);
static bool IsNonConstInput(const ge::Node& node, const size_t index = 0UL);
static bool IsNonConstInput(const ge::ConstNodePtr& node, const size_t index = 0UL);
static std::vector<ge::GeTensorDesc> GetNonConstTensorDesc(const ge::ConstNodePtr& node);
static graphStatus AddConstOpToAnchor(const InDataAnchorPtr in_anchor, const GeTensorPtr& tensor_ptr);
static Operator CreateOperatorFromOpDesc(OpDescPtr op_desc);
static Operator CreateOperatorFromNode(ge::ConstNodePtr node_ptr);
static OpDescPtr GetOpDescFromOperator(const Operator& oprt);
static graphStatus CopyOperatorLinks(const std::map<std::string, ge::Operator> &src_op_list,
std::map<std::string, ge::Operator> &dst_op_list);
static graphStatus CopyOperators(const ComputeGraphPtr &dst_compute_graph,
const std::map<ConstNodePtr, NodePtr> &node_old_2_new,
const std::map<ConstOpDescPtr, OpDescPtr> &op_desc_old_2_new,
const std::map<std::string, ge::Operator> &src_op_list,
std::map<std::string, ge::Operator> &dst_op_list);
static OpDescPtr CloneOpDesc(const ConstOpDescPtr &org_op_desc);
__attribute__((weak)) static OpDescPtr CopyOpDesc(const ConstOpDescPtr &org_op_desc);
__attribute__((weak)) static OpDescPtr CreateConstOp(const GeTensorPtr& tensor_ptr);
static OpDescPtr CreateConstOp(const GeTensorPtr& tensor_ptr, const bool copy);
static OpDescPtr CreateConstOpZeroCopy(const GeTensorPtr& tensor_ptr);
static graphStatus SetSubgraphInstanceName(const std::string &subgraph_name,
const std::string &subgraph_instance_name, OpDescPtr &op_desc);
static ConstGeTensorBarePtr GetInputConstData(const Operator &op, const uint32_t idx);
static void SetRuntimeContextToOperator(const Operator &op, RuntimeInferenceContext *const context);
static void SetCallbackGetConstInputFuncToOperator(const Operator &op,
GetConstInputOnRuntimeFun get_const_input_func);
static bool HasCallbackGetConstInputFunc(const Operator &op);
static std::map<size_t, std::pair<size_t, size_t>> GetInputIrIndexes2InstanceIndexesPairMap(const OpDescPtr &op_desc);
static std::map<size_t, std::pair<size_t, size_t>> GetOutputIrIndexes2InstanceIndexesPairMap(
const OpDescPtr &op_desc);
static graphStatus GetIrInputInstanceDescRange(const OpDescPtr &op,
std::map<size_t, std::pair<size_t, size_t>> &ir_input_2_range);
__attribute__((weak)) static graphStatus GetIrInputRawDescRange(const OpDescPtr &op,
std::map<size_t, std::pair<size_t, size_t>> &ir_input_2_range);
static graphStatus GetIrOutputDescRange(const OpDescPtr &op,
std::map<size_t, std::pair<size_t, size_t>> &ir_output_2_range);
static ge::graphStatus GetInputIrIndexByInstanceIndex(const OpDescPtr &op_desc,
size_t instance_index, size_t &ir_index);
static ge::graphStatus GetOutputIrIndexByInstanceIndex(const OpDescPtr &op_desc,
size_t instance_index, size_t &ir_index);
static ge::graphStatus GetInstanceNum(const OpDescPtr &op_desc, size_t ir_index, size_t start_index,
const std::map<uint32_t, std::string> &valid_index_2_names,
size_t &instance_num);
static ge::graphStatus GetPromoteIrInputList(const OpDescPtr &op_desc,
std::vector<std::vector<size_t>> &promote_index_list);
static ge::graphStatus GetPromoteInstanceInputList(const OpDescPtr &op_desc,
std::vector<std::vector<size_t>> &promote_index_list);
static ge::graphStatus GetIrInputDtypeSymIds(const OpDescPtr &op_desc, std::vector<std::string> &dtype_sym_ids);
static ge::graphStatus GetIrOutputDtypeSymIds(const OpDescPtr &op_desc, std::vector<std::string> &dtype_sym_ids);
private:
static GeTensorPtr MutableWeights(ge::OpDesc& op_desc);
static GeTensorPtr MutableWeights(const ge::OpDescPtr op_desc);
static graphStatus SetWeights(ge::OpDesc& op_desc, const GeTensorPtr weight);
static graphStatus SetWeights(ge::OpDescPtr op_desc, const GeTensorPtr weight);
};
}
#endif
