fastvideo.platforms.rocm

fastvideo.platforms.rocm#

This file is a platform abstraction for ROCm GPUs, adjusted to match the structure and interface of cuda.py.

Module Contents#

Classes#

RocmPlatform

Data#

logger

API#

class fastvideo.platforms.rocm.RocmPlatform[source]#

Bases: fastvideo.platforms.interface.Platform

device_control_env_var: str[source]#

‘CUDA_VISIBLE_DEVICES’

device_name: str[source]#

‘rocm’

device_type: str[source]#

‘cuda’

dispatch_key: str[source]#

‘CUDA’

classmethod get_attn_backend_cls(selected_backend: fastvideo.platforms.interface.AttentionBackendEnum | None, head_size: int, dtype: torch.dtype) → str[source]#
classmethod get_current_memory_usage(device: torch.device | None = None) → float[source]#
classmethod get_device_capability(device_id: int = 0) → fastvideo.platforms.interface.DeviceCapability[source]#
classmethod get_device_communicator_cls() → str[source]#
classmethod get_device_name(device_id: int = 0) → str[source]#
classmethod get_device_total_memory(device_id: int = 0) → int[source]#
classmethod is_async_output_supported(enforce_eager: bool | None) → bool[source]#
classmethod log_warnings() → None[source]#
fastvideo.platforms.rocm.logger[source]#

‘init_logger(…)’

Contents