函数:stop_dump_args
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
x |
|
x |
|
x |
功能说明
调用本接口关闭算子信息统计功能,并需与函数:start_dump_args接口配合使用,将算子信息文件输出到path参数指定的目录,一个shape对应一个算子信息文件,文件中包含算子类型、算子属性、算子输入&输出的format/数据类型/shape等信息。
使用场景:例如要统计某个模型执行涉及哪些算子,可在模型执行之前调用acl.op.start_dump_args接口,在模型执行之后调用acl.op.stop_dump_args接口,接口调用成功后,在path参数指定的目录下生成每个算子shape的算子信息文件。
函数原型
- C函数原型
1
aclError aclopStopDumpArgs(uint32_t dumpType)
- python函数
1
ret = acl.op.stop_dump_args(dump_type)
参数说明
参数名 |
说明 |
---|---|
dumpType |
int,指定dump信息的类型。 当前仅支持ACL_OP_DUMP_OP_AICORE_ARGS,表示统计所有算子信息。 ACL_OP_DUMP_OP_AICORE_ARGS 0x00000001 |
返回值说明
返回值 |
说明 |
---|---|
ret |
int,错误码,返回0表示成功,返回其它值表示失败。 |
约束说明
仅支持在单算子API执行场景下使用本接口,否则无法生成dump文件。
Atlas 200I/500 A2推理产品,不支持该接口。
父主题: 单算子模型执行