- 头文件:#include <ge/ge_api.h>
- 库文件:libge_runner.so
用户将自己的Allocator注册给GE,适用于使用用户的内存池场景。
此接口需要配合、或接口使用,并且需要在LoadGraph接口调用前注册。
[object Object]
对于同一条流,多次调用本接口,以最后一次注册为准。
对于不同流,如果用户使用同一个Allocator,不可以多条流并发执行,在执行下一条Stream前,需要对上一Stream做流同步。
将Allocator中的内存释放给操作系统前,需要先调用“aclrtSynchronizeStream”接口执行流同步,确保Stream中的任务已执行完成。
接口详细介绍请参见《Runtime运行时 API》中的“Stream管理”。