ATLAS Offline Software
Loading...
Searching...
No Matches
ActsGPUMagFieldConfig Namespace Reference

Functions

ComponentAccumulator CUDAMagFieldProviderToolCfg (flags, name="CUDAMagFieldProviderTool", **kwargs)
ComponentAccumulator HIPMagFieldProviderToolCfg (flags, name="HIPMagFieldProviderTool", **kwargs)
ComponentAccumulator DeviceMagFieldProviderToolCfg (flags, **kwargs)
ComponentAccumulator JSONDeviceMagFieldProviderSvcCfg (flags, **kwargs)

Function Documentation

◆ CUDAMagFieldProviderToolCfg()

ComponentAccumulator CUDAMagFieldProviderToolCfg ( flags,
name = "CUDAMagFieldProviderTool",
** kwargs )

Definition at line 11 of file ActsGPUMagFieldConfig.py.

13 **kwargs) -> ComponentAccumulator:
14 acc = ComponentAccumulator()
15
16 kwargs.setdefault("MagFieldStorage","global_memory")
17 acc.setPrivateTools(
18 CompFactory.ActsTrk.CUDAMagFieldProviderTool(name, **kwargs))
19 return acc
20

◆ DeviceMagFieldProviderToolCfg()

ComponentAccumulator DeviceMagFieldProviderToolCfg ( flags,
** kwargs )

Definition at line 31 of file ActsGPUMagFieldConfig.py.

32 **kwargs) -> ComponentAccumulator:
33 acc = ComponentAccumulator()
34
35 if flags.Device.Backend is DeviceBackend.CUDA:
36 acc.setPrivateTools(acc.popToolsAndMerge(CUDAMagFieldProviderToolCfg(flags)))
37
38 elif ((flags.Device.Backend is DeviceBackend.HIPAMD) or (flags.Device.Backend is DeviceBackend.HIPNVIDIA)):
39 acc.setPrivateTools(acc.popToolsAndMerge(HIPMagFieldProviderToolCfg(flags)))
40 else:
41 raise ValueError(f"Unsupported device backend: {flags.Acts.DeviceBackend}")
42
43 return acc
44

◆ HIPMagFieldProviderToolCfg()

ComponentAccumulator HIPMagFieldProviderToolCfg ( flags,
name = "HIPMagFieldProviderTool",
** kwargs )

Definition at line 21 of file ActsGPUMagFieldConfig.py.

23 **kwargs) -> ComponentAccumulator:
24 acc = ComponentAccumulator()
25
26 kwargs.setdefault("MagFieldStorage","global_memory")
27 acc.setPrivateTools(
28 CompFactory.ActsTrk.HIPMagFieldProviderTool(name, **kwargs))
29 return acc
30

◆ JSONDeviceMagFieldProviderSvcCfg()

ComponentAccumulator JSONDeviceMagFieldProviderSvcCfg ( flags,
** kwargs )

Definition at line 45 of file ActsGPUMagFieldConfig.py.

45def JSONDeviceMagFieldProviderSvcCfg(flags, **kwargs) -> ComponentAccumulator:
46
47 acc = ComponentAccumulator()
48
49 kwargs.setdefault("MagFieldFile", "dev/ACTS/detray-itk/ITk_bfield.cvf")
50 kwargs.setdefault("DeviceMagFieldObjectName", "TracccDeviceMagField")
51 kwargs.setdefault("HostMagFieldObjectName", "TracccHostMagField")
52
53 kwargs.setdefault("DeviceMagFieldProviderTool", acc.popToolsAndMerge(DeviceMagFieldProviderToolCfg(flags)))
54
55 svc = CompFactory.ActsTrk.JSONDeviceMagFieldProviderSvc(**kwargs)
56 acc.addService(svc, primary=True, create=True)
57 return acc
58
59
60