类说明
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。
属性名 |
属性类型 |
默认值 |
属性说明 |
---|---|---|---|
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 |
None |
算子核函数源码信息。 |
父主题: DebugOp类