UpdateGraphFeatureMemoryBase
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
头文件/库文件
- 头文件：#include <ge/ge_api_v2.h>
- 库文件：libge_runner_v2.so
功能说明
用于更新Graph的Feature内存基址。
在调用本接口前，必须先调用CompileGraph接口进行图编译；内存大小从GetCompiledGraphSummary>GetFeatureMemorySize接口中获取。
Feature内存指的是模型执行过程中所需要的中间内存（比如中间节点的输入输出等内存）。
函数原型
Status UpdateGraphFeatureMemoryBase(uint32_t graph_id, const void *const memory, size_t size)参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| graph_id | 输入 | Graph对应的ID。 |
| memory | 输入 | 设置的Feature内存基地址。 |
| size | 输入 | 设置的Feature内存大小。 |
返回值说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| - | Status | SUCCESS：设置成功。 FAILED：设置失败。 |
约束说明
- 调用CompileGraph完成图编译后，您可以通过GetCompiledGraphSummary接口获取Feature地址是否可刷新，只有Feature地址可刷新的图才支持重复调用本接口刷新Feature地址。
- 该接口只适用于静态编译图，可以通过GetCompiledGraphSummary接口中的IsStatic接口获取图是否为静态编译。
- 若使用了本接口，又配置了RegisterExternalAllocator接口，则RegisterExternalAllocator接口不生效。
- 不能与SetGraphFixedFeatureMemoryBaseWithType接口入参为MEMORY_TYPE_DEFAULT同时使用。
- 不能与UpdateGraphRefreshableFeatureMemoryBase同时使用。
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
