將C++/CUDA類傳遞給PyCUDA的SourceModule

Question

我有一個用C++編寫的類，它也使用cuda_runtime.h中的一些定義，這是來自名爲ADOL-C的開源項目的一部分，您可以看看here！

這在我使用CUDA-C時有效，但我想以某種方式在PyCUDA中導入此類，如果有可能的話。因此，我將在內核中使用這個類（不是在'main'中）來定義用於計算函數派生的特定變量。有沒有辦法將這個類傳遞給PyCUDA的SourceModule？

我問了一個類似的問題，但在這裏我想解釋更多一點。因此，有一個解決方案使用nvcc -cubin（感謝talonmies）編譯我的C代碼，然後用driver.module_from_file（）導入它，但是，我想使用SourceModule並將這些內核寫入.py文件，所以它可能更加用戶友好。我的例子會是這個樣子：

from pycuda import driver, gpuarray 
from pycuda.compiler import SourceModule 
import pycuda.autoinit 
kernel_code_template=""" 
__global__ void myfunction(float* inx, float* outy, float* outderiv) 
{ 
    //defining thread index 
    ... 
    //declare dependent and independet variables as adoubles 
    //this is a part of my question 
    adtl::adouble y[3]; 
    adtl::adouble x[3]; 
    // ... 
} 
""" 

...這只是一個想法，但SourceModule不會知道什麼是「adouble的」，因爲它們在類定義adoublecuda.h定義的，所以我希望你現在能更好地理解我的問題。有沒有人有我應該從哪裏開始的線索？如果沒有，我將在CUDA-C中編寫這些內核，並使用nvcc -cubin選項。

感謝您的幫助！

Answer 1

PyCUDA SourceModule系統實際上只是一種將代碼傳遞到文件中的方法，將nvcc文件編譯爲cubin文件並（可選）將該文件加載到當前CUDA上下文中。 PyCUDA編譯器模塊對CUDA內核語法或代碼一無所知，並且對已編譯的代碼幾乎沒有影響[幾乎限定符是因爲它可以將用戶提交的代碼用extern "C" { }聲明括起來以停止C++符號的修改]。

所以，我們要做什麼，我認爲你是問，你應該只需要一個#include語句無論頭設備代碼提交的字符串的需要，並在通過include_dirs通過Python列表一套合適的搜索路徑 關鍵字選項。如果你這樣做：

from pycuda import driver, gpuarray 
from pycuda.compiler import SourceModule 
import pycuda.autoinit 
kernel_code_template=""" 

#include "adoublecuda.h" 
__global__ void myfunction(float* inx, float* outy, float* outderiv) 
{ 
    //defining thread index 
    ... 
    //declare dependent and independet variables as adoubles 
    //this is a part of my question 
    adtl::adouble y[3]; 
    adtl::adouble x[3]; 
    // ... 
} 

""" 

module = SourceModule(kernel_code_template, include_dirs=['path/to/adoublecuda']) 

它應該自動工作（注意未經測試，使用風險自負）。