---

CUDAUtil.cu

Go to the documentation of this file.

/***************************************************************************
 *cr
 *cr            (C) Copyright 1995-2008 The Board of Trustees of the
 *cr                        University of Illinois
 *cr                         All Rights Reserved
 *cr
 ***************************************************************************/

/***************************************************************************
 * RCS INFORMATION:
 *
 *      $RCSfile: CUDAUtil.cu,v $
 *      $Author: johns $        $Locker:  $             $State: Exp $
 *      $Revision: 1.8 $        $Date: 2008/06/01 09:19:01 $
 *
 ***************************************************************************
 * DESCRIPTION:
 *   CUDA API wrapper for use by the CUDAAccel C++ class 
 ***************************************************************************/
#include <string.h>

#if defined(__cplusplus)
extern "C" {
#endif

int vmd_cuda_num_devices(int *numdev) {
  if (cudaGetDeviceCount(numdev) != cudaSuccess) {
    *numdev = 0;
    return -1;
  }

  // Check for emulation mode 'device', and ignore it if found
  if (*numdev == 1) {
    cudaDeviceProp deviceProp;
    if (cudaGetDeviceProperties(&deviceProp, 0) != cudaSuccess) {
      *numdev = 0;
      return -1;
    }

    // disregard emulation mode as unusable for our purposes
    if (((deviceProp.major < 1) && (deviceProp.minor < 1)) ||
        ((deviceProp.major == 9999) && (deviceProp.minor == 9999))) {
      *numdev = 0;
    }
  }

  return 0;
}

int vmd_cuda_device_props(int dev, char *name, int namelen,
                          int *revmajor, int *revminor, 
                          unsigned long *memb, int *clockratekhz,
                          int *smcount, int *overlap) {
  cudaDeviceProp deviceProp;
  if (cudaGetDeviceProperties(&deviceProp, dev) != cudaSuccess)
    return -1;

  strncpy(name, deviceProp.name, namelen);
  *revmajor = deviceProp.major;
  *revminor = deviceProp.minor;
  *memb = deviceProp.totalGlobalMem;
  *clockratekhz = deviceProp.clockRate;
#if CUDART_VERSION >= 2000
  *smcount = deviceProp.multiProcessorCount;
  *overlap = (deviceProp.deviceOverlap != 0);
#else
  *smcount = -1;
  *overlap = 0; 
#endif

  return 0;
}

#if defined(__cplusplus)
}
#endif

---