在内核编译时检查 OpenCL 供应商 ID？

Question

在 OpenCL kernel 中，如果针对 NVIDIA GPU，我想做一件事，否则我想做另一件事，即

#if MAGIC_HERE
asm("some ptx code");
#else
// some OpenCL C code which does the same thing, sort of
#endif

我应该放什么而不是MAGIC_HERE ？

Answer 1

2个选项：

• 使用特定于 nvidia 的扩展之一，它始终存在于 Nvidia GPU 上，但不存在于其他供应商的 GPU 上，例如MAGIC_HERE = cl_nv_pragma_unroll
• 将#define MAGIC_HERE嵌入运行时编译的 OpenCL C 代码字符串中 C++，当且仅当cl_device.getInfo<CL_DEVICE_VENDOR>()包含NVIDIA