Skip to content

rocm

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

Classes

fastvideo.platforms.rocm.RocmPlatform

Bases: Platform

Functions

fastvideo.platforms.rocm.RocmPlatform.get_torch_device classmethod 
get_torch_device()

Return torch.cuda

Source code in fastvideo/platforms/rocm.py 
61
62
63
64
65
66
@classmethod
def get_torch_device(cls):
    """
    Return torch.cuda
    """
    return torch.cuda

Functions