dragon.infrastructure.gpu_desc

Module for detecting GPU devices across differet vendors

Functions

apply_gpu_affinity(env, device_list, env_str)
find_accelerators()	Scan for accelerators across all supported vendors

Classes

AcceleratorDescriptor

AcceleratorDescriptor(vendor: dragon.accelerators.AccVendor = <AccVendor.UNKNOWN: 4>, device_list: list[int] = <factory>, env_str: str = '')

class AcceleratorDescriptor

Bases: object

AcceleratorDescriptor(vendor: dragon.accelerators.AccVendor = <AccVendor.UNKNOWN: 4>, device_list: list[int] = <factory>, env_str: str = ‘’)

vendor: AccVendor = 4

device_list: list[int]

env_str: str = ''

get_sdict()

classmethod from_sdict(sdict)

__init__(vendor: AccVendor = AccVendor.UNKNOWN, device_list: list[int] = <factory>, env_str: str = '') → None

apply_gpu_affinity(env, device_list, env_str, inherited_env=None)

find_accelerators() → AcceleratorDescriptor

Scan for accelerators across all supported vendors