昇腾社区首页
中文
注册

beta)torch_npu.npu.set_dump

接口原型

torch_npu.npu.set_dump(path_to_json)

功能描述

传入配置文件来配置dump参数。

参数说明

path_to_json:配置文件所在的路径,包含文件名,用户需根据实际情况配置。具体配置请参考《CANN 应用开发接口》中“函数:set_dump”章节

支持的型号

  • Atlas 训练系列产品
  • Atlas A2 训练系列产品
  • Atlas A3 训练系列产品
  • Atlas 推理系列产品

调用示例

1
>>>torch_npu.npu.set_dump("/home/HwHiAiUser/dump.json")