Skip to content

如何使用Tiling依赖的头文件

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

  • 使用标准C++语法定义Tiling结构体

    这种方式需要引入依赖的头文件如下。所有高阶API的Tiling结构体定义在AscendC::tiling命名空间下,因此需要通过AscendC::tiling访问具体API的Tiling结构体。

    Text
    #include "kernel_tiling/kernel_tiling.h"
    
    // ...
    AscendC::tiling::TCubeTiling cubeTilingData;
    
  • 使用TILING_DATA_DEF宏定义Tiling结构体

    这种方式需要引入依赖的头文件如下。所有高阶API的Tiling结构体和Tiling函数定义在optiling命名空间下。

    Text
    #include "tiling/tiling_api.h"
    
    namespace optiling {
    // ...
    }
    

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区