TAPsCUDA_GlobalFns_Def.cu

Go to the documentation of this file.

/******************************************************************************
TAPsCUDA_GlobalFns_Def.cu

CUDA Global Functions Definition (i.e., cpp file).

SUKITTI PUNAK   (08/27/2008)
UPDATE          (09/17/2009)
******************************************************************************/

#include "TAPsCUDA_GlobalFns.cu"

BEGIN_NAMESPACE_TAPs__CUDA
//=============================================================================
//-----------------------------------------------------------------------------

// Initialize CUDA Device
void Host__Init ( int argc, char **argv )
{
    printf( "START: CUT_DEVICE_INIT\n" );
    CUT_DEVICE_INIT(argc, argv);
    printf( "END:   CUT_DEVICE_INIT\n" );
}

// Initialize CUDA (with Page-Locked Host Memory)
void Host__Init_withPLHM ( int argc, char **argv )
{
    if ( !SetDevice( 0 ) ) {
        printf( "FAILED: setting the device 0! -- File: %s Line: %i\n", __FILE__, __LINE__ );
        exit(-501);
    }
    int device;
    if ( !GetDevice( device ) ) {
        printf( "FAILED: getting the current device! -- File: %s Line: %i\n", __FILE__, __LINE__ );
        exit(-501);
    }
    if ( !CanDeviceMapHostMemory( device ) ) {
        printf( "FAILED: the current device cannot map host memory! -- File: %s Line: %i\n", __FILE__, __LINE__ );
        exit(-502);
    }
    cudaError_t err = cudaSetDeviceFlags( cudaDeviceMapHost );
    if ( err != cudaSuccess ) {
        printf( "FAILED: cannot set device's map host memory flag! (%s) -- File: %s Line: %i\n", cudaGetErrorString( err ), __FILE__, __LINE__ );
        exit(-503);
    }
}

// Get number of devices
int DeviceCount ()
{
    int count;
    cudaGetDeviceCount( &count );
    return count;
}

// Get the device on which the active host thread executes the device code
bool GetDevice ( int & device )
{
    cudaError_t err = cudaGetDevice( & device );
    if ( err == cudaSuccess ) return true;
    printf( "FAILED: GetDevice(%i) -- %s\n", device, GetErrorString( err ) );
    return false;
}

// Set the device on which the active host thread executes the device code
// If the host thread has already initialized the CUDA runtime by calling 
// non-device management runtime functions, this call results in 
// cudaErrorSetOnActiveProcess.
bool SetDevice ( int device )
{
    cudaError_t err = cudaSetDevice( device );
    if ( err == cudaSuccess ) return true;
    printf( "FAILED: SetDevice(%i) -- %s\n", device, GetErrorString( err ) );
    return false;
}

// Get Device's Properties
bool GetDeviceProperties ( cudaDeviceProp * prop, int device )
{
    cudaError_t err = cudaGetDeviceProperties( prop, device );
    if ( err == cudaSuccess ) return true;
    printf( "FAILED: GetDeviceProperties( ..., %i) -- %s\n", device, GetErrorString( err ) );
    return false;
}

// Can Map Host Memory (i.e. can utilize CUDA's Page-Locked Host Memory)
bool CanDeviceMapHostMemory ( int device )
{
    //if ( !SetDevice( device ) )   return false;
    cudaDeviceProp prop;
    if ( !GetDeviceProperties( &prop, device ) )    return false;
    return prop.canMapHostMemory == 1;
}




// Check error and exit the program if the action flag is true
bool CheckError ( cudaError_t error, const char * msg, bool bExitProgram )
{
    if ( error == cudaSuccess ) return true;
    if ( msg ) {
        printf ( "ERROR: %s -- %s\n", msg, cudaGetErrorString( error ) );
    }
    else {
        printf ( "ERROR: %s\n", cudaGetErrorString( error ) );
    }
    if ( bExitProgram ) exit( -555 );
    return false;
}

// Convert cudaError_t to const char *
const char * GetErrorString ( cudaError_t error )
{
    return cudaGetErrorString( error );
}

//-----------------------------------------------------------------------------
//=============================================================================
END_NAMESPACE_TAPs__CUDA
//-----------------------------------------------------------------------------
//34567890123456789012345678901234567890123456789012345678901234567890123456789
//--+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----