device.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.device

Functions
	aquin.compute.device._import_torch ()
bool	aquin.compute.device.cuda_available ()
bool	aquin.compute.device.mps_available ()
bool	aquin.compute.device.is_rocm_build ()
str	aquin.compute.device.cuda_backend_label ()
str|None	aquin.compute.device._env_override ()
str	aquin.compute.device.resolve_compute_device (*, str|None prefer=None)
Any	aquin.compute.device.default_dtype_for_device (str|None device=None)
tuple[str, Any]	aquin.compute.device.device_and_dtype (*, str|None prefer=None)
bool	aquin.compute.device.accelerator_available ()
bool	aquin.compute.device.allow_cpu_load ()
None	aquin.compute.device.empty_device_cache (str|None device=None)
None	aquin.compute.device.synchronize_device (str|None device=None)
bool	aquin.compute.device.is_oom_error (BaseException exc)
list[dict[str, Any]]	aquin.compute.device._cuda_gpu_entries ()
bool	aquin.compute.device._is_windows ()
str|None	aquin.compute.device._rocm_install_hint ()
str|None	aquin.compute.device.torch_version_label ()
bool	aquin.compute.device.torch_is_cpu_wheel ()
list[str]	aquin.compute.device.nvidia_smi_gpu_names ()
str|None	aquin.compute.device._nvidia_cpu_torch_hint ()
str|None	aquin.compute.device._best_fix_hint ()
str|None	aquin.compute.device._windows_amd_hint ()
str|None	aquin.compute.device._darwin_cpu_hint ()
str	aquin.compute.device.load_blocked_message (str model_id)
str	aquin.compute.device.require_load_device (str model_id, dict[str, Any]|None cfg=None)
dict[str, Any]	aquin.compute.device.probe_backend ()
str	aquin.compute.device.format_backend_summary ()
str	aquin.compute.device.probe_backend_json ()

Variables
	aquin.compute.device._VALID_OVERRIDES = frozenset({"cuda", "mps", "cpu", "auto"})