编译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 |