CUDAConfigFlags Namespace Reference

Classes
class	CUDAStream

Functions
	createCUDAConfigFlags ()

Function Documentation

createCUDAConfigFlags()

createCUDAConfigFlags

(

)

Function to create the flags for using CUDA devices/GPUs in Athena

Definition at line 22 of file CUDAConfigFlags.py.

def createCUDAConfigFlags():
    '''Function to create the flags for using CUDA devices/GPUs in Athena
    '''
 
    # Create the flags container.
    result = AthConfigFlags()
 
    # Memory (resource) management flags.
    result.addFlag('CUDA.Stream', CUDAStream.Single, type=CUDAStream,
                   help='Strategy for selecting/providing a CUDA stream')
 
    # Return the container.
    return result