codebase / stimlib

  #ifndef STIM_CUDA_DEVICES
  #define STIM_CUDA_DEVICES

  
  #include <cuda.h>
  

  namespace stim{

  	extern "C"
  	int maxThreadsPerBlock(){
  		int device;
  		cudaGetDevice(&device);		//get the id of the current device
  		cudaDeviceProp props;		//device property structure
  		cudaGetDeviceProperties(&props, device);
  		return props.maxThreadsPerBlock;
  	}

  	extern "C"
  	size_t sharedMemPerBlock(){
  		int device;
  		cudaGetDevice(&device);		//get the id of the current device
  		cudaDeviceProp props;		//device property structure
  		cudaGetDeviceProperties(&props, device);
  		return props.sharedMemPerBlock;
  	}

  	extern "C"
  	size_t constMem(){
  		int device;
  		cudaGetDevice(&device);		//get the id of the current device
  		cudaDeviceProp props;		//device property structure
  		cudaGetDeviceProperties(&props, device);
  		return props.totalConstMem;
  	}
  
  	//tests that a given device ID is valid and provides at least the specified compute capability
  	bool testDevice(int d, int major, int minor){
  		int nd;
  		cudaGetDeviceCount(&nd);		//get the number of CUDA devices

  		if(d <= nd && d >= 0)	{		//if the given ID has an associated device

  			cudaDeviceProp props;
  			cudaGetDeviceProperties(&props, d);	//get the device properties structure

  			if(props.major > major){
  				return true;
  			}
  			else if(props.major == major && props.minor >= minor){
  				return true;
  			}

  		}
  		return false;
  	}
  
  	//tests each device ID in a list and returns the number of devices that fit the desired
  	//	compute capability
  	int testDevices(int* dlist, unsigned n_devices, int major, int minor){
  		int valid = 0;

  		for(unsigned d = 0; d < n_devices; d++){

  			if(testDevice(dlist[d], major, minor))
  				valid++;
  		}
  		return valid;
  	}

  
  	void printDevice(int device){
  		int nd;
  		cudaGetDeviceCount(&nd);		//get the number of CUDA devices
  		printf("CUDA Device Diagnosis: [%i]\n", device);
  		if(device < 0){
  			printf("Device %i is an invalid device ID\n", device);
  		}
  		else if(device >= nd){
  			printf("Device %i is unavailable - only %i devices are detected", device, nd);
  		}
  		else{
  			cudaDeviceProp props;
  			cudaGetDeviceProperties(&props, device);	//get the device properties structure
  			printf("compute capability: %i.%i\n", props.major, props.minor);
  		}
  	}

  }	//end namespace rts
  

  #endif