在使用数学函数时在openCl和device中使用函数

Question

所以我要说我有以下代码：

//MyFile.h 
void MyFunction;

//MyFile.cpp
#include "MyFile.h"

void MyFunction()
{
    float foo = sqrtf(85.3);
}

现在，我正在使用的实际函数更复杂，但重点是它们使用sqrtf（），sin（）等函数。

我想在主机和我的内核中使用这个功能，但我不想两次定义它，因为这似乎是不好的做法。 现在我尝试使用clCreateProgramWithSource（）方法将“MyFile.h”传递给我的cl_program。

如果我在设备上运行它，它会工作正常，因为openCL内置了数学函数。 但是，如果我希望它在主机上运行，我需要包含cmath或其他库以定义sqrt，sin等。但是如果我包含cmath以便它在主机上工作，它将不再能够在设备上运行，因为openCL不能包含/ import cmath。

所以我似乎无法找到一种方法来定义它并在两个地方使用它。 如果主机不依赖于cmath来获得这些数学函数，那么这不是问题，但目前我不知道该怎么做。

Answer 1

您可以测试OpenCL C特定预处理器宏的存在（或不存在），以选择性地包含仅主机头文件。

例如：

#ifndef __OPENCL_C_VERSION__
  #include <cmath>
#endif

Answer 2

那么你需要的是添加条件语句以包括在主机上编译它。 有#ifdef/#ifndef预处理器宏将检查是否存在定义并根据此测试采取操作

#ifndef __OPENCL_C_VERSION__

// I am host, so include <cmath>
  #include <cmath>

#else

// I am device
// ...

#endif