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', '${chip_version}')
注意事项
在调用调测API前,先调用本接口生成算子基本信息管理类对象。
父主题: 公共接口