visual-c++ - 如何将 cudaStreamAddCallback() 与类成员方法一起使用？

Question

我正在尝试使用 cudaStreamAddCallback() 来同步我的 cuda 例程，但我无法实现它，也是因为文档并不明确。
cuda-C-programming-guide 说回调必须定义为:

void CUDART_CB MyCallback(void *data){}

并且正在谈论需要设置的 cudaStreamCallbackBlocking 之类的标志；
而 Cuda_Toolhit_Reference_Manual 和 cuda_runtime_api.h 需要回调的其他实现:

void CUDART_CB MyCallback (cudaStream_t stream, cudaError_t status, void *userData){}

并提到该标志供将来使用并且需要 0 作为参数。
此外，调用函数如下:

cudaStreamAddCallback(GpuStream, MyCallback, &BufSwitchParams, 0);

并使用 VS 2010 尝试为 64 位编译我收到消息:“void(__stdcall CMyClass::*)(cudaStream_t stream, cudaError_t status, void *userData)”类型的参数与“cudaStreamCallback_t”类型的参数不兼容.

有人已经实现了这个功能并且能够帮助我摆脱困境，同时在这里发布一个片段？

Answer 1

您将一个类方法传递给 cudaStreamAddCallback ，但它应该是一个非成员函数(全局或静态)。如果要使用类方法，则应实现将调用该方法的包装函数:

class MyClass
{
public:
    static void CUDART_CB Callback(cudaStream_t stream, cudaError_t status, void *userData);

private:
    void callbackFunc();
};

void CUDART_CB MyClass::Callback(cudaStream_t stream, cudaError_t status, void *userData)
{
    MyClass* thiz = (MyClass*) userData;
    thiz->callbackFunc();
}

void MyClass::callbackFunc()
{
    // implementation here
}

MyClass* obj = new MyClass;
cudaStreamAddCallback(GpuStream, MyClass::Callback, obj, 0);