下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

GetCompiledGraphSummary

函数功能

查询图编译结果的概要信息。包括Feature内存大小、Const内存大小、Stream、Event数目及内存是否可刷新等信息。您可以根据该信息,自行申请内存,再通过UpdateGraphFeatureMemoryBaseSetGraphConstMemoryBase接口设置Feature内存、Const内存基址。

函数原型

CompiledGraphSummaryPtr GetCompiledGraphSummary(uint32_t graph_id);

参数说明

参数名

输入/输出

描述

graphId

输入

子图对应的id。

返回值

参数名

类型

描述

-

CompiledGraphSummaryPtr

图编译结果的概要信息CompiledGraphSummary的share_ptr。

CompiledGraphSummary具体结构如下所示:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
class GE_FUNC_VISIBILITY CompiledGraphSummary {
 public:
  class Builder;
  class SummaryData;
  CompiledGraphSummary();
  CompiledGraphSummary &operator=(const CompiledGraphSummary &) & = delete;
  CompiledGraphSummary(const CompiledGraphSummary &) = delete;
  // 返回graph是否是静态编译的,只有在静态编译的情况下,才可以使用后续的获取Summary信息的接口
  bool IsStatic() const;
  // 获取编译后的Const内存大小
  Status GetConstMemorySize(size_t &size) const;
  // 获取编译后的Feature内存大小
  Status GetFeatureMemorySize(size_t &size) const;
  // 获取Feature基址是否可刷新
  Status GetFeatureMemoryBaseRefreshable (bool &v) const;
 
  // 获取Stream数目
  Status GetStreamNum(size_t &num) const;
  // 获取Event数目
  Status GetEventNum(size_t &num) const;
  // 获取图编译的输出shape,输出shape可用于计算outputs所占内存大小
  Status GetOutputShapes(std::vector<ge::Shape> &shapes) const;
 private:
  CompiledGraphSummary() = default;
  std::shared_ptr<SummaryData> data_{nullptr};
};

约束说明

在调用本接口前,必须先调用CompileGraph接口进行图编译。

产品支持情况

Atlas 推理系列产品(Ascend 310P处理器),支持

Atlas 训练系列产品,支持

Atlas A2训练系列产品,支持

Atlas 200/300/500 推理产品,不支持

Atlas 200/500 A2推理产品,不支持

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词