文档
注册
评分
提单
论坛
小AI

create_debug_op

函数功能

构造算子信息对象。

函数原型

def create_debug_op(op_type: str, core_type: str, chip_version: str) -> DebugOp

参数说明

参数名

输入/输出

说明

op_type

输入

字符串类型,算子名称,例如'AddCustom'。

core_type

输入

字符串类型,算子core类型,请根据算子实际情况设置。可选值为Aicore、MixCore、CubeCore、VectorCore、MixVectorCore,MixVectorCore为预留参数,开发者无需关注。

chip_version

输入

昇腾AI处理器的版本。

若为“”或None,工具会自动获取,也支持手动设置,在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的值为Ascendxxxyy

返回值

返回DebugOp类型的实例对象。

使用说明

import ascendebug
debug_op = ascendebug.create_debug_op('OpCustom', 'VectorCore', 'Ascend310P')

注意事项

在调用调测API前,先调用本接口生成算子基本信息管理类对象。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词