ARTICLE DETAIL

资讯详情

深耕网站建设、视觉设计与SEO优化的一线实战洞察。

CANN/ge流分配节点获取API

CANN/ge流分配节点获取API GEStreamAllocationSummaryGetAllNodes产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√头文件/库文件头文件#include ge/ge_graph_compile_summary.h库文件libge_compiler.so功能说明获取根图和子图的所有节点。函数原型ge::Status GEStreamAllocationSummaryGetAllNodes(const ge::CompiledGraphSummary compiled_graph_summary, std::mapAscendString, std::vectorstd::vectorGNode graph_to_all_nodes);参数说明参数输入/输出说明compiled_graph_summary输入图编译后的概要信息。graph_to_all_nodes输出map格式key为图名称value为所有节点的向量其中索引表示逻辑流ID。返回值说明参数名类型说明-ge::StatusSUCCESS接口调用成功。FAILED接口调用失败约束说明无创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表