下载
中文
注册

AICPU

用法

python3 aicpu_generate_model.py --cores <core_num> -p <process_id> -t <npu_type>

参数名称

<core_num>:昇腾AI处理器AI Core的个数,默认为“2”。(预留参数,暂不使用)

<process_id>:批量生成算子多进程调度的进程ID,默认值为“0”,无需设置。

<npu_type>:硬件形态,当前<npu_type>支持Atlas 200/300/500 推理产品以及Atlas 推理系列产品,取值范围分别为:310、310P,默认为“310”

说明

执行此命令,用户可以得到一组算子模型文件。

AICPU算子模型文件只需生成一次,会全部生成所有算法的算子。