fastvideo.platforms.mps#

Module Contents#

Classes#

Data#

API#

class fastvideo.platforms.mps.MpsPlatform[source]#

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.

fastvideo.platforms.mps.logger[source]#

โ€˜init_logger(โ€ฆ)โ€™