NpuCompileInfo
ascendebug提供了NpuCompileInfo类,用于存储与NPU编译相关的信息,包括是否使能打印功能、核函数运行core数分配方式等信息。
具体定义如下:
@dataclass class NpuCompileInfo(): # 编译属性info dump_mode: str = "" syncall: bool = False task_ration: List[int] = field(default_factory=list)
关于类成员的详细说明请参见表1。
属性名 |
属性类型 |
默认值 |
属性说明 |
---|---|---|---|
dump_mode |
str |
"" |
printf/PRINTF/DumpTensor/DumpAccChkPoint/assert/时间戳打印功能的模式配置,与CompileNpuOptions.dump_mode配套使用。
|
syncall |
bool |
False |
硬同步功能的使能开关,仅适用于Atlas A2训练系列产品/Atlas 800I A2推理产品。
说明:
|
task_ration |
List[int] |
[] |
函数运行core数分配方式,格式为[n1, n2],例如task_ration=[1, 0], 表示“1:0”的分配方式。默认的[]表示使用系统默认格式。
说明:
|
父主题: 基本类