文档
注册

类说明

ascendebug提供了DebugOp算子信息管理类,保存管理算子类型、算子输入/输出描述信息、算子核函数信息等。

具体定义如下:

class DebugOp:
    self.op_type: str = ''
    self.core_type: str = ''
    self.chip_version: str = ''
    self.param_args: List[Any] = None          # TensorDesc, TensorListDesc, WorkspaceDesc, TilingDesc
    self.attrs: List[AttrDesc] = None
    self.kernel_info: KernelInfo = None

关于类成员的详细说明请参见表1

表1 DebugOp类成员说明

属性名

属性类型

默认值

属性说明

op_type

str

""

算子名。

core_type

str

Aicore

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

chip_version

str

""

昇腾AI处理器的版本。

若为“”或None,工具会自动获取,具体取值如下:
  • Atlas 推理系列产品(Ascend 310P处理器):如AscendxxxP1、AscendxxxP3等。
  • Atlas A2训练系列产品/Atlas 800I A2推理产品:如AscendxxxB1、AscendxxxB2、AscendxxxB3、AscendxxxB4等。

param_args

list

[]

算子核函数入参描述信息列表。

attrs

list

[]

算子属性描述信息列表。

kernel_info

OpKernelInfo

None

算子核函数源码信息。

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

当前产品无相关内容

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