Kernel函数

如异构编程QuickStartDemoACL.cce所示，Kernel函数是由__global__关键字标注的特殊函数，是设备函数的入口，由Host侧通过特定的异构调用语法<<<>>>执行。一个cce文件可包含若干个Kernel函数，Host代码既可以和Device代码共存于同一文件中，也可以分开存放，基本符合常规C++函数调用习惯。

// moduleA.cce

// Kernel 函数
__global__ [aicore] void foo(/* Args */...) {}
__global__ [aicore] void bar(/* Args */...) {}

RetCode_t FooAPI(/* Args */ ...) {
  foo<<<blockDim, nullptr, stream>>>(...);
}


// moduleB.cce

extern __global__ [aicore] void bar(/*Args*/ ...);
extern RetCode_t FooAPI(...);

int main() {
  bar<<<...>>>(...);
  FooAPI(...);
}