| 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") |
|
|
