昇腾社区首页
中文
注册

预定义宏

毕昇编译器如其他语言一样,会提供一些内置的宏方便用户编写程序。这些宏在compiler包中的头文件中可以看到。以下表格介绍一些常用的宏及其说明。

表1 毕昇编译器内置常用的预定义宏

宏名

说明

__NPU_ARCH__

2201

只在Device侧定义,表示芯片架构编号,2201对应A2/A3

__DAV_VEC__

不涉及

只在Device侧定义,表示VEC核

__DAV_CUBE__

不涉及

只在Device侧定义,表示CUBE核

__global__

__attribute__((cce_kernel))

只在Host和Device侧都定义,只能在Device侧使用,Kernel函数的标识符

__ubuf__

__attribute__((cce_unif_buff))

只在Host和Device侧都定义,只能在Device侧使用,表示ubuf地址空间

__fbuf__

__attribute__((cce_fixpipe_buff))

只在Host和Device侧都定义,只能在Device侧使用,表示fbuf地址空间