acldumpRegCallback
函数功能
Dump数据回调函数注册接口。
aclmdlInitDump接口、acldumpRegCallback接口(通过该接口注册的回调函数需由用户自行实现,回调函数实现逻辑中包括获取Dump数据及数据长度)、acldumpUnregCallback接口、aclmdlFinalizeDump接口配合使用,用于通过回调函数获取Dump数据。场景举例如下:
- 执行一个模型,通过回调获取Dump数据:
支持以下两种方式:
- 在aclInit接口处不启用模型Dump配置、单算子Dump配置
aclInit接口-->aclmdlInitDump接口-->acldumpRegCallback接口-->模型加载-->模型执行-->acldumpUnregCallback接口-->aclmdlFinalizeDump接口-->模型卸载-->aclFinalize接口
- 在aclInit接口处启用模型Dump配置、单算子Dump配置,在aclInit接口处启用Dump配置时需配置落盘路径,但如果调用了acldumpRegCallback接口,则落盘不生效,以回调函数获取的Dump数据为准
aclInit接口-->acldumpRegCallback接口-->模型加载-->模型执行-->acldumpUnregCallback接口-->模型卸载-->aclFinalize接口
- 在aclInit接口处不启用模型Dump配置、单算子Dump配置
- 执行两个不同的模型,通过回调获取Dump数据,该场景下,只要不调用acldumpUnregCallback接口取消注册回调函数,则可通过回调函数获取两个模型的dump数据:
aclInit接口-->aclmdlInitDump接口-->acldumpRegCallback接口-->模型1加载-->模型1执行-->-->模型2加载-->模型2执行-->acldumpUnregCallback接口-->aclmdlFinalizeDump接口-->模型卸载-->aclFinalize接口
函数原型
aclError acldumpRegCallback(int32_t (* const messageCallback)(const acldumpChunk *, int32_t len), int32_t flag)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
messageCallback |
输入 |
回调函数指针,用于接收回调数据的回调。
|
flag |
输入 |
在调用回调接口后是否还落盘dump数据:
|
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: 模型执行