CUDA_獲取指定設備

請求計算能力2.0設備(費米架構):編程

struct cudaDeviceProp device_prop;
int chosen_device;

memeset(device_prop, 0, sizeof(cudaDeviceProp));
device_prop.major = 2;
device_prop.minor = 0;

if(cudaChooseDevice(&chosen_device, device_prop) != cudaErrorInvalidValue)
{
	CUDA_CALL(cudaSetDevice(chosen_device));
}

cuda錯誤處理:架構

每一個函數返回一個錯誤代碼時,每次函數調用必須進行錯誤檢查,並編寫處理函數。這使得錯誤處理很麻煩,而且致使高度縮進的程序代碼:函數

if (cudaMalloc(..) == cudaSuccess) 
{
	if (cudaEventCreate(&event) == cudaSuccess)
	{
		
	}
}
else
{
	...
}

爲了重複性的編程,可使用下面的宏來調用CUDA應用程序接口:code

#define CUDA_CALL(x) {const cudaError_t a=(x); if (a != cudaSuccess) {printf("\nCUDAError:%s(err_num=%d)\n",cudaGetErrorString(a),a); cudaDeviceReset(); assert(0);}}
相關文章
相關標籤/搜索