mirror of
https://github.com/LTTLabsOSS/markbench-tests.git
synced 2026-01-08 05:33:52 -05:00
65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
"""Modules for executing shell commands and parsing strings"""
|
|
import subprocess
|
|
import re
|
|
|
|
def get_winml_devices(procyon_path):
|
|
"""
|
|
Function which uses the ProcyonCmd.exe to list all available winml devices on the system. Returns a dictionary of device names and IDs
|
|
"""
|
|
|
|
# run the command line utility for procyon in order to list available winml devices
|
|
winml_devices = subprocess.run([f'{procyon_path}', '--list-winml-devices'], shell=True, capture_output=True, text=True, check=True).stdout
|
|
|
|
winml_devices_split = winml_devices.split('\n')
|
|
winml_devices_parsed = [device[9::] for device in winml_devices_split if re.search(r"(amd|nvidia|intel)", device.lower())]
|
|
unique_winml_devices = list(dict.fromkeys(winml_devices_parsed))
|
|
winml_dict = {device_split.split(', ')[0]:device_split.split(', ')[1] for device_split in unique_winml_devices}
|
|
|
|
return winml_dict
|
|
|
|
def get_openvino_devices(procyon_path):
|
|
"""
|
|
Function which uses the ProcyonCmd.exe to list all available openvino devices on the system. Returns a dictionary of device type and name
|
|
"""
|
|
|
|
openvino_devices = subprocess.run([f'{procyon_path}', '--list-openvino-devices'], shell=True, capture_output=True, text=True, check=True).stdout
|
|
|
|
openvino_devices_split = openvino_devices.split('\n')
|
|
openvino_devices_parsed = [device[9::] for device in openvino_devices_split if re.search(r"(amd|nvidia|intel)", device.lower())]
|
|
unique_openvino_devices = list(dict.fromkeys(openvino_devices_parsed))
|
|
openvino_dict = {device_split.split(', ')[1]:device_split.split(', ')[0] for device_split in unique_openvino_devices}
|
|
|
|
return openvino_dict
|
|
|
|
def get_openvino_gpu(openvino_devices, gpu_id):
|
|
"""
|
|
function which checks the openvino_devices dictionary for GPU entries.
|
|
If only one gpu is detected, there should not be a GPU.0 and GPU.1 entry
|
|
so we just return the first detected gpu regardless of requested ID
|
|
"""
|
|
|
|
gpu = openvino_devices.get("GPU", None)
|
|
|
|
if gpu is None:
|
|
gpu = openvino_devices.get(gpu_id, "No Openvino GPU Detected")
|
|
|
|
return gpu
|
|
|
|
def get_cuda_devices(procyon_path):
|
|
"""
|
|
Function which uses the ProcyonCmd.exe to list all available openvino devices on the system. Returns a dictionary of device type and name
|
|
"""
|
|
|
|
cuda_devices = subprocess.run([f'{procyon_path}', '--list-cuda-devices'], shell=True, capture_output=True, text=True, check=True).stdout
|
|
|
|
cuda_devices_split = cuda_devices.split('\n')
|
|
cuda_devices_parsed = [device[9::] for device in cuda_devices_split if re.search(r"(nvidia)", device.lower())]
|
|
unique_cuda_devices = list(dict.fromkeys(cuda_devices_parsed))
|
|
|
|
if len(unique_cuda_devices) > 0:
|
|
cuda_dict = {device_split.split(', ')[1]:device_split.split(', ')[0] for device_split in unique_cuda_devices}
|
|
else:
|
|
cuda_dict = {}
|
|
|
|
return cuda_dict
|