函数:stop_dump_args
C函数原型 |
aclError aclopStopDumpArgs(uint32_t dumpType) |
---|---|
Python函数 |
ret = acl.op.stop_dump_args(dump_type) |
函数功能 |
调用本接口关闭算子信息统计功能,并需与函数:start_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 |
返回值说明 |
返回0表示成功,返回其它值表示失败。 |
约束说明 |
仅支持在单算子API执行场景下使用本接口,否则无法生成dump文件。 Atlas 200/300/500 推理产品,不支持该接口。 Atlas 推理系列产品,不支持该接口。 Atlas 200I/500 A2推理产品,不支持该接口。 Atlas 训练系列产品,不支持该接口。 Atlas A2训练系列产品/Atlas 800I A2推理产品,不支持该接口。 |
父主题: 单算子模型执行