76 lines
3.0 KiB
Plaintext
76 lines
3.0 KiB
Plaintext
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
//===============================================================================================================================================================================================================200
|
||
|
|
// SET_DEVICE CODE
|
||
|
|
//===============================================================================================================================================================================================================200
|
||
|
|
|
||
|
|
//======================================================================================================================================================150
|
||
|
|
// INCLUDE/DEFINE
|
||
|
|
//======================================================================================================================================================150
|
||
|
|
|
||
|
|
#include "cuda.h" // (in library path specified to compiler)
|
||
|
|
|
||
|
|
//======================================================================================================================================================150
|
||
|
|
// FUNCTIONS
|
||
|
|
//======================================================================================================================================================150
|
||
|
|
|
||
|
|
//====================================================================================================100
|
||
|
|
// SET DEVICE
|
||
|
|
//====================================================================================================100
|
||
|
|
|
||
|
|
void setdevice(void){
|
||
|
|
|
||
|
|
// variables
|
||
|
|
int num_devices;
|
||
|
|
int device;
|
||
|
|
|
||
|
|
// work
|
||
|
|
cudaGetDeviceCount(&num_devices);
|
||
|
|
if (num_devices > 1) {
|
||
|
|
|
||
|
|
// variables
|
||
|
|
int max_multiprocessors;
|
||
|
|
int max_device;
|
||
|
|
cudaDeviceProp properties;
|
||
|
|
|
||
|
|
// initialize variables
|
||
|
|
max_multiprocessors = 0;
|
||
|
|
max_device = 0;
|
||
|
|
|
||
|
|
for (device = 0; device < num_devices; device++) {
|
||
|
|
cudaGetDeviceProperties(&properties, device);
|
||
|
|
if (max_multiprocessors < properties.multiProcessorCount) {
|
||
|
|
max_multiprocessors = properties.multiProcessorCount;
|
||
|
|
max_device = device;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
cudaSetDevice(max_device);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
//====================================================================================================100
|
||
|
|
// GET LAST ERROR
|
||
|
|
//====================================================================================================100
|
||
|
|
|
||
|
|
void checkCUDAError(const char *msg)
|
||
|
|
{
|
||
|
|
cudaError_t err = cudaGetLastError();
|
||
|
|
if( cudaSuccess != err) {
|
||
|
|
// fprintf(stderr, "Cuda error: %s: %s.\n", msg, cudaGetErrorString( err) );
|
||
|
|
printf("Cuda error: %s: %s.\n", msg, cudaGetErrorString( err) );
|
||
|
|
fflush(NULL);
|
||
|
|
exit(EXIT_FAILURE);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//===============================================================================================================================================================================================================200
|
||
|
|
// END
|
||
|
|
//===============================================================================================================================================================================================================200
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|