npu

Classes

fastvideo.platforms.npu.NPUPlatform

Bases: Platform

Functions

fastvideo.platforms.npu.NPUPlatform.get_torch_device classmethod

get_torch_device()

Return torch.npu

Source code in fastvideo/platforms/npu.py

@classmethod
def get_torch_device(cls):
    """
    Return torch.npu
    """
    return torch.npu

Functions