otx.backend.native.accelerators

Lightning accelerator for XPU device.

Classes

XPUAccelerator()

Support for a XPU, optimized for large-scale machine learning.

class otx.backend.native.accelerators.XPUAccelerator

Bases: Accelerator

Support for a XPU, optimized for large-scale machine learning.

static auto_device_count() → int

Returns number of XPU devices available.

get_device_stats(device: str | device) → dict[str, Any]

Returns XPU devices stats.

static get_parallel_devices(devices: list) → list[device]

Generates a list of parrallel devices.

static is_available() → bool

Checks if XPU available.

static parse_devices(devices: str | list | device) → list

Parses devices for multi-GPU training.

setup_device(device: device) → None

Sets up the specified device.

teardown() → None

Clean up any state created by the accelerator.