昇腾社区首页
中文
注册

编译AscendNPU IR

bishengir-compile可以将AscendNPU IR编译成可以在昇腾AI处理器上执行的二进制文件。

编译选项

常用的编译选项说明如下:

选项

是否必需

说明

--help

查看帮助。

-o <file>

指定输出文件的名称和位置。

--enable-triton-kernel-compile

使能Triton相关的编译优化。输入的IR需要为通过triton-shared工具转换得到的MLIR。

--enable-hfusion-compile

使能HFusion方言层级的编译优化。

--enable-hivm-compile

使能HIVM方言层级的编译优化。默认为开。

--enable-auto-multi-buffer

使能昇腾AI处理器片上Buffer自动乒乓多缓冲优化。

编译命令

编译过程中,输入input.mlir经过HFusion、HIVM方言编译优化后,生成二进制文件outputFile.o。

编译命令如下:

1
bishengir-compile --enable-hfusion-compile=true --enable-hivm-compile=true input.mlir -o outputFile