下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

编译与运行样例

环境准备

请注意单算子API接口的编译依赖开发套软件包Ascend-cann-toolkit与算子二进制软件包Ascend-cann-kernels。

编译前准备

本章以开发和运行环境合设场景为例,即带昇腾AI处理器的机器既作为开发环境又作为运行环境。该场景下,代码开发与代码运行在同一台机器上。

这里以Add算子为例,其他算子的调用逻辑和编译脚本类似,请根据实际情况准备算子API调用脚本(*.cpp)和编译脚本(CMakeLists)。

  • 示例代码

    已知Add算子实现了张量加法运算,计算公式为:y=x1+αx2。您可以从aclnnAdd&aclnnInplaceAdd中“调用示例”获取示例代码,并将代码文件命名为“test_add.cpp”。

  • CMakeLists文件

    Camke文件示例如下,请根据实际情况修改:

    # CMake lowest version requirement
    cmake_minimum_required(VERSION 3.14)
    
    # 设置工程名
    project(ACLNN_EXAMPLE)
    
    # Compile options
    add_compile_options(-std=c++11)
    
    # 设置编译选项
    set(CMAKE_RUNTIME_OUTPUT_DIRECTORY  "./bin")    
    set(CMAKE_CXX_FLAGS_DEBUG "-fPIC -O0 -g -Wall")
    set(CMAKE_CXX_FLAGS_RELEASE "-fPIC -O2 -Wall")
    
    # 设置可执行文件名(如opapi_test),并指定待运行算子文件*.cpp所在目录
    add_executable(opapi_test
                   test_add.cpp) 
    
    # 设置ASCEND_PATH(CANN软件包目录,请根据实际路径修改)和INCLUDE_BASE_DIR(头文件目录)
    if(NOT "$ENV{ASCEND_CUSTOM_PATH}" STREQUAL "")      
        set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})
    else()
        set(ASCEND_PATH "/usr/local/Ascend/ascend-toolkit/latest")
    endif()
    set(INCLUDE_BASE_DIR "${ASCEND_PATH}/include")
    include_directories(
        ${INCLUDE_BASE_DIR}
        ${INCLUDE_BASE_DIR}/aclnn
    )
    
    # 设置链接的库文件路径
    target_link_libraries(opapi_test PRIVATE
                          ${ASCEND_PATH}/lib64/libascendcl.so
                          ${ASCEND_PATH}/lib64/libnnopbase.so
                          ${ASCEND_PATH}/lib64/libopapi.so)
    
    # 可执行文件在CMakeLists文件所在目录的bin目录下
    install(TARGETS opapi_test DESTINATION ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})

编译与运行

  1. 提前准备好算子的调用代码(*.cpp)和编译脚本(CMakeLists.txt)。

  2. 配置环境变量。

    安装CANN软件后,使用CANN运行用户(如HwHiAiUser)登录环境,执行如下命令设置环境变量。其中${install_path}为CANN软件安装后文件存储路径,请根据实际情况替换该路径。

    source ${install_path}/set_env.sh
  3. 编译并运行。

    • 进入CMakeLists.txt所在目录,执行如下命令,新建build目录存放生成的编译文件。

      mkdir -p build 
    • 进入build目录,执行cmake命令编译,再执行make命令生成可执行文件。

      cd build
      cmake ../ -DCMAKE_CXX_COMPILER=g++ -DCMAKE_SKIP_RPATH=TRUE
      make

      编译成功后,会在build目录的bin文件夹下生成opapi_test可执行文件。

    • 进入bin目录,运行可执行文件opapi_test。

      cd bin
      ./opapi_test

      以Add算子的运行结果为例,运行后的结果示例如下:

      result[0] is: 1.200000
      result[1] is: 2.200000
      result[2] is: 3.200000
      result[3] is: 5.400000
      result[4] is: 6.400000
      result[5] is: 7.400000
      result[6] is: 9.600000
      result[7] is: 10.600000
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词