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

UpdateGraphFeatureMemoryBase

函数功能

用于更新Graph的Feature内存基址。Featrure内存指的是模型执行过程中所需要的中间内存(比如中间节点的输入输出等内存)。

函数原型

Status UpdateGraphFeatureMemoryBase(uint32_t graph_id, const void *const memory, size_t size);

参数说明

参数名

输入/输出

描述

graph_id

输入

子图对应的id。

memory

输入

设置的feature内存基地址。

size

输入

设置的feature内存大小。

返回值

参数名

类型

描述

-

Status

SUCCESS:设置成功。

FAILED:设置失败。

约束说明

  • 在调用本接口前,必须先调用CompileGraph接口进行图编译。
  • 调用CompileGraph完成图编辑后,您可以通过GetCompiledGraphSummary接口获取feature地址是否可刷新,只有feature地址可刷新的图才支持继续调用本接口刷新feature地址。
  • 若使用了本接口,又配置了RegisterExternalAllocator接口, 则RegisterExternalAllocator接口不生效。

产品支持情况

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

Atlas 训练系列产品,支持

Atlas A2训练系列产品,支持

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

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

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

当前产品无相关内容

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