下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

类说明

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,工具会自动获取,也支持手动设置,查询方法如下:

在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的值为Ascendxxxyy

param_args

list

[]

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

attrs

list

[]

算子属性描述信息列表。

kernel_info

OpKernelInfo

None

算子核函数源码信息。

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

当前产品无相关内容

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