昇腾社区首页
中文
注册

异构编译步骤

毕昇编译器支持异构编译,当需要编写包含异构程序时,可按如下步骤进行:

  1. 从CANN发布包中获取毕昇编译器程序(bisheng)的路径,并设置环境变量。${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。
    # 设置编译器环境变量,如:
    $export PATH=${INSTALL_DIR}/compiler/ccec_compiler/bin:$PATH
  2. 代码开发。编写异构程序的基本流程如下:
    1. include必需的Runtime(运行时管理)头文件。
    2. 通过Runtime API创建devicestream
    3. 申请设备内存,按需将输入数据拷贝至设备内存。
    4. 编写设备kernel函数。
    5. 通过<<<>>>内核调用符运行设备kernel函数。
    6. 按需将输出数据拷贝至主机内存,并释放设备内存。
  3. 获取runtime的动态链接库以及头文件的位置。在本示例中,编译依赖和所在位置如下:

    编译依赖

    位置

    以下简称

    runtime动态链接库

    ${INSTALL_DIR}/lib64/

    $RT_LIB

    runtime头文件

    ${INSTALL_DIR}/include

    $RT_INC

  4. 编写并执行编译命令,一步编译生成可执行文件。下文场景示例为开发环境和运行环境一致的情况。
    // 简化写法,无意义场景不可运行,仅演示编译使用方法
    #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
  5. 运行可执行文件。
     # 运行: 
     $export LD_LIBRARY_PATH=$RT_LIB:$LD_LIBRARY_PATH
     $./axpy