Bases: fastvideo.platforms.interface.Platform
-
device_control_env_var: str[source]
‘MPS_VISIBLE_DEVICES’
-
device_name: str[source]
‘mps’
-
device_type: str[source]
‘mps’
-
dispatch_key: str[source]
‘MPS’
-
classmethod get_attn_backend_cls(selected_backend: fastvideo.platforms.AttentionBackendEnum | None, head_size: int, dtype: torch.dtype) → str[source]
-
classmethod get_current_memory_usage(device: torch.types.Device | None = None) → float[source]
-
abstract classmethod get_device_capability(device_id: int = 0) → fastvideo.platforms.interface.DeviceCapability | None[source]
-
classmethod get_device_communicator_cls() → str[source]
-
abstract classmethod get_device_name(device_id: int = 0) → str[source]
-
abstract classmethod get_device_total_memory(device_id: int = 0) → int[source]
-
abstract classmethod get_device_uuid(device_id: int = 0) → str[source]
-
classmethod is_async_output_supported(enforce_eager: bool | None) → bool[source]
-
classmethod seed_everything(seed: int | None = None) → None[source]
Set the seed for MPS device.
