CUDA內核可以調用cublas函數嗎？

Question

我知道它聽起來很奇怪，但這裏是我的場景：

我需要做矩陣矩陣乘法（A（n * k）* B（k * n）），但我只需要對角線元素對輸出矩陣進行評估。我搜索cublas圖書館並沒有發現任何可以做到這一點的2級或3級功能。 因此，我決定將A的每一行和B的每一列分配到CUDA線程中。對於每個線程（idx），我需要計算點積「A [idx，：] * B [：，idx]」並將其保存爲相應的對角線輸出。現在因爲這個dot產品也需要一些時間，我想知道我是否可以在這裏調用cublas函數（比如cublasSdot）來實現它。

如果我錯過了一些cublas函數可以直接實現我的目標（只計算矩陣 - 矩陣乘法的對角元素），這個問題可以被丟棄。

Answer 1

是的，它可以。

「CUDA C/C++中提供的語言界面和Device Runtime API是CUDA Runtime API的一個子集，CUDA Runtime API的語法和語義保留在設備上以便於可以在主機或設備環境中運行的API例程輕鬆實現代碼重用內核還可以直接調用GPU庫，例如CUBLAS，而無需返回CPU。 Source

Here你可以看到和使用CUDA和CUBLAS庫函數cublasSgemv矩陣向量乘法。

Answer 2

確保您使用設備庫來調用cublas。您不能使用您用來從主機調用它的相同庫;有關使用cuda設備庫的詳細信息可以在cuda工具包中找到：http://docs.nvidia.com/cuda/cublas/index.html#device-api

查看7_CUDALibraries /下的cuda 5示例。