cudaStreamAddCallback() yöntemini sınıf üyesi yöntemiyle nasıl kullanabilirim?

Question

cudaStreamAddCallback() öğesini kullanarak cuda rutinimi senkronize etmeye çalışıyorum, ancak belgelerin açık olmaması nedeniyle bunu uygulayamıyorum. cuda-C-programlama-rehber geri arama olarak tanımlanan gerektiğini söylüyor:

void CUDART_CB MyCallback(void *data){} 

ve ayarlanması gerekir cudaStreamCallbackBlocking gibi bayrakları bahsediyor; Cuda_Toolhit_Reference_Manual ve cuda_runtime_api.h geri arama bir diğer uygulama gerektirirken:

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

ve bayrak ileride kullanılmak üzere olduğunu belirterek bağımsız değişken olarak bir 0 gerektirir. Ayrıca, aşağıdaki gibi işlevini çağırarak:

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

ve iletisini alıyorum 64bit için derlemeye çalışırken VS 2010 kullanıyorsanız çalışma: Boşluğun”türü argüman (__ stdcall CMyClass :: *) (cudaStream_t akışı, cudaError_t status, void * userData) ”," cudaStreamCallback_t "türünde bir parametreyle uyumsuz.

Birisi bu işlevi zaten uygulamış ve burada bir pasaj yayınlarken ikilemden kurtulmama yardımcı olabilir mi?

Answer 1

Bir sınıf yöntemini cudaStreamAddCallback'a geçirirsiniz, ancak üye olmayan bir işlev (global veya statik) olmalıdır.

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);