gpu_select#

class GPUSelect#

Bases: object

canary_gpu_backend_detect(config: Config) str | None#
canary_gpu_list_gpus(config: Config) list[dict] | None#
canary_addoption(parser: Parser) None#
canary_addhooks(pluginmanager: CanaryPluginManager) None#
canary_addconfig(config: Config) None#
detect_gpu_backend(config: Config) None#
canary_fill_gpu(config: canary.Config, pool: ResourcePool) None#

Fill local-node GPU resources from the selected GPU backend.

GPU backend discovery is local-node discovery, so this hook populates the first node’s resources. Multi-node/HPC plugins can replace or populate nodes separately.

get_plugin_name(plugin: Any) str#