昇腾社区首页
中文
注册
开发者
下载

如何使用Tiling依赖的头文件

由于AI处理器的Scalar计算单元执行能力有限,为减少算子Kernel侧的Scalar计算,将部分计算在Host端执行,这需要编写Host端Tiling代码。注意,在程序中调用高阶API的Tiling接口或者使用高阶API的Tiling结构体参数时,需要引入依赖的头文件。在不同的Tiling实现方式下,具体为:

  • 使用标准C++语法定义Tiling结构体
    这种方式需要引入依赖的头文件如下。所有高阶API的Tiling结构体定义在AscendC::tiling命名空间下,因此需要通过AscendC::tiling访问具体API的Tiling结构体。
    1
    2
    3
    4
    #include "kernel_tiling/kernel_tiling.h"
    
    // ...
    AscendC::tiling::TCubeTiling cubeTilingData;
    
  • 使用TILING_DATA_DEF宏定义Tiling结构体
    这种方式需要引入依赖的头文件如下。所有高阶API的Tiling结构体和Tiling函数定义在optiling命名空间下。
    1
    2
    3
    4
    5
    #include "tiling/tiling_api.h"
    
    namespace optiling {
    // ...
    }