backend/device.py

import platform
from constants import DEVICE
import torch
import openvino as ov

core = ov.Core()


def is_openvino_device() -> bool:
    if DEVICE.lower() == "cpu" or DEVICE.lower()[0] == "g":
        return True
    else:
        return False


def get_device_name() -> str:
    if DEVICE == "cuda" or DEVICE == "mps":
        default_gpu_index = torch.cuda.current_device()
        return torch.cuda.get_device_name(default_gpu_index)
    elif platform.system().lower() == "darwin":
        return platform.processor()
    elif is_openvino_device():
        return core.get_property(DEVICE.upper(), "FULL_DEVICE_NAME")