mstar.utils.device

Classes

DeviceCapability(device)

Query one device's capabilities without initializing its hardware.

class mstar.utils.device.DeviceCapability(device)

Bases: object

Query one device’s capabilities without initializing its hardware.

Runtime operations use PyTorch’s generic stream, event, and accelerator APIs directly. Kernel dispatch remains with the resource that owns it.

Parameters:

device (torch.device)

property is_accelerator: bool

property is_available: bool

is_capturing()

Return type:

bool

static of(device)

Parameters:

device (device | str)

Return type:

DeviceCapability

property pin_memory: bool