函数:start_dump_args
C函数原型 |
aclError aclopStartDumpArgs(uint32_t dumpType, const char *path) |
---|---|
Python函数 |
ret = acl.op.start_dump_args(dump_type, path) |
函数功能 |
调用本接口开启算子信息统计功能,并需与函数:stop_dump_args接口配合使用,将算子信息文件输出到path参数指定的目录,一个shape对应一个算子信息文件,文件中包含算子类型、算子属性、算子输入&输出的format/数据类型/shape等信息。 使用场景:例如要统计某个模型执行涉及哪些算子,可在模型执行之前调用acl.op.start_dump_args接口,在模型执行之后调用acl.op.stop_dump_args接口,接口调用成功后,在path参数指定的目录下生成每个算子shape的算子信息文件。 |
输入说明 |
dump_type:指定dump信息的类型。 当前仅支持ACL_OP_DUMP_OP_AICORE_ARGS,表示统计所有算子信息。 ACL_OP_DUMP_OP_AICORE_ARGS = 0x00000001 path:指定dump文件的保存路径,支持绝对路径或相对路径(指相对应用可执行文件所在的目录),但用户需确保路径存在或者该路径可以被创建。 |
返回值说明 |
返回0表示成功,返回其它值表示失败。 |
约束说明 |
仅支持在单算子API执行场景下使用本接口,否则无法生成dump文件。 Atlas 200/300/500 推理产品,不支持该接口。 Atlas 推理系列产品,不支持该接口。 Atlas 200I/500 A2推理产品,不支持该接口。 Atlas 训练系列产品,不支持该接口。 Atlas A2训练系列产品/Atlas 800I A2推理产品,不支持该接口。 |
父主题: 单算子模型执行