ATLAS Offline Software
Loading...
Searching...
No Matches
python.DeviceConfigFlags Namespace Reference

Classes

class  DeviceBackend

Functions

 createDeviceConfigFlags ()

Function Documentation

◆ createDeviceConfigFlags()

python.DeviceConfigFlags.createDeviceConfigFlags ( )
Function to create the flags for using accelerator devices in Athena

Definition at line 27 of file DeviceConfigFlags.py.

27def createDeviceConfigFlags():
28 '''Function to create the flags for using accelerator devices in Athena
29 '''
30
31 # Create the flags container.
32 result = AthConfigFlags()
33
34 # Memory (resource) management flags.
35 result.addFlag('Device.Memory.Cache', True,
36 help='Whether or not to cache memory allocations')
37 result.addFlag('Device.Memory.Debug', False,
38 help='Whether or not to actively debug memory allocations')
39 result.addFlag('Device.Memory.Shared', False,
40 help='Whether or not to use "shared" memory for device '
41 'allocations. In case it is set to False, a separate '
42 'host and device memory resource is to be used.')
43
44 # Memory copy management flag(s).
45 result.addFlag('Device.Copy.Async', True,
46 help='Whether or not to use asynchronous memory copies')
47
48 # Backend selection flag(s).
49 result.addFlag('Device.Backend', DeviceBackend.CUDA, type=DeviceBackend,
50 help='The backend to use for accelerator devices')
51
52 # Return the container.
53 return result