CUDA編程——設備屬性的使用

根據(jù)cudaGetDeviceCount()和cudaGetDeviceProperties()中返回的結果,我們可以對每個設備進行操作。

代碼如下:

int main(void) {

????cudaDeviceProp prop;

????int dev;

????cudaGetDevice(&dev);

????printf("ID of current CUDA device: %d\n", dev);

????memset(&prop, 0, sizeof(cudaDeviceProp));

????prop.major = 1;

????prop.minor = 3;

????cudaChooseDevice(&dev, &prop);

????printf("ID of CUDA device closest to revision 1.3: %d\n", dev);

????cudaSetDevice(dev);

????return 0;

}

聲明一個cudaDeviceProp結構后,將其傳遞給cudaChooseDevice(),CUDA運行時將查找是否存在某個設備滿足這些條件,cudaChooseDevice()函數(shù)將返回一個設備ID,然后將這個ID傳遞給cudaSetDevice(),隨后所有的設備操作都將在這個設備上執(zhí)行。

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

友情鏈接更多精彩內容