Question

是否可以在OpenCL代码中定义自己的函数，以便内核可以调用它们？是的，我在哪里可以看到一些简单的例子？

Answer 1

用于创建程序的函数是......

cl_program clCreateProgramWithSource  (     
    cl_context context,
    cl_uint count,
    const char **strings,
    const size_t *lengths,
    cl_int *errcode_ret)

您可以将功能放在字符串参数中，如下所示

float AddVector(float a, float b)
{
    return a + b;
}

kernel void VectorAdd(
    global read_only float* a,
    global read_only float* b,
    global write_only float* c )
{
    int index = get_global_id(0);
    //c[index] = a[index] + b[index];
    c[index] = AddVector(a[index], b[index]);
}

现在您有一个用户定义的函数“AddVector”和一个内核函数“VectorAdd”

Answer 2

基于代码示例here，您可以编写如下函数：

inline int add(int a,int b)
{
   return a+b;
}

（例如，查看DXTC中的.cl文件或bitonic排序示例。）

我不知道这是否只是一个nvidia扩展，但OpenCL文档谈到了“辅助功能”以及内核。

Answer 3

OpenCL支持辅助功能。有关示例，请参见this link的第19页。

Answer 4

我用Google搜索了一下，然后不断回到这个问题:-P

最后，我所做的是使用宏，因为内联依赖于实现，并且宏在c99 OpenCL程序的上下文中似乎没有任何主要的缺点？例如：

#define getFilterBoardOffset( filter, inputPlane ) \
    ( ( filter * gInputPlanes + inputPlane ) * gFilterSizeSquared )
#define getResultBoardOffset( n, filter ) \
    ( ( n * gNumFilters + filter ) * gOutputBoardSizeSquared )

而不是：

inline float getFilterBoardOffset( float filter, int inputPlane ) { 
    return ( filter * gInputPlanes + inputPlane ) * gFilterSizeSquared; 
}
inline float getResultBoardOffset( float n, int filter ) { 
    return ( n * gNumFilters + filter ) * gOutputBoardSizeSquared; 
}

OpenCL用户定义的内联函数

4 个答案: