--soc_version
功能说明
当--op_tune_mode=fast,并且执行aoe命令的环境没有Device时,需要通过--soc_version指定昇腾AI处理器的版本。
当前该功能仅支持Atlas A2训练系列产品。
关联参数
只有当--op_tune_mode=fast时可使用。
参数取值
参数值:<soc_version>
如果无法确定具体的<soc_version>,则在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的<soc_version>值为Ascendxxxyy。
参数值约束:请使用与芯片名相对应的<soc_version>取值进行模型转换,然后再进行推理。
示例
--op_tune_file=/home/lxl/json --op_tune_mode=fast --precision_mode=must_keep_origin_dtype --op_tune_kernel_path=/home/lxl/kernel --soc_version=AscendxxxB1
父主题: 高级功能