异构编译步骤
毕昇编译器支持异构编译,当需要编写包含异构程序时,可按如下步骤进行:
- 从CANN发布包中获取毕昇编译器程序(bisheng)的路径,并设置环境变量。${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。
# 设置编译器环境变量,如: $export PATH=${INSTALL_DIR}/compiler/ccec_compiler/bin:$PATH
- 代码开发。编写异构程序的基本流程如下:
- include必需的Runtime(运行时管理)头文件。
- 通过Runtime API创建device和stream。
- 申请设备内存,按需将输入数据拷贝至设备内存。
- 编写设备kernel函数。
- 通过<<<>>>内核调用符运行设备kernel函数。
- 按需将输出数据拷贝至主机内存,并释放设备内存。
- 获取runtime的动态链接库以及头文件的位置。在本示例中,编译依赖和所在位置如下:
编译依赖
位置
以下简称
runtime动态链接库
${INSTALL_DIR}/lib64/
$RT_LIB
runtime头文件
${INSTALL_DIR}/include
$RT_INC
- 编写并执行编译命令,一步编译生成可执行文件。下文场景示例为开发环境和运行环境一致的情况。
// 简化写法,无意义场景不可运行,仅演示编译使用方法 #ifdef ASCENDC_CPU_DEBUG #define __aicore__ #else #define __aicore__ [aicore] #endif __global__ __aicore__ void foo(__gm__ int* buf) { *buf = 1; } int main(int argc, char* argv[]) { int a[100]; // nullptr - No L2 usage here foo<<<5, nullptr, nullptr>>>(a); return 0; }
编译命令如下:
# 功能:Host & Device代码混合编译,生成可执行文件 axpy # 编译命令: $bisheng --cce-soc-version=AscendXXXYY --cce-soc-core-type=VecCore -O2 axpy.cce -o axpy -I$RT_INC -L$RT_LIB
- 运行可执行文件。
# 运行: $export LD_LIBRARY_PATH=$RT_LIB:$LD_LIBRARY_PATH $./axpy
父主题: 异构编译