文档
注册

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,工具会自动获取,也支持手动设置,取值如下:
  • Atlas 推理系列产品(Ascend 310P处理器):如Ascend310P1、Ascend310P3等。
  • Atlas A2训练系列产品/Atlas 800I A2推理产品:如AscendxxxB1、AscendxxxB2、AscendxxxB3、AscendxxxB4等。

返回值

返回DebugOp类型的实例对象。

使用说明

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

注意事项

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

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

当前产品无相关内容

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