github/markbench-tests
1
1
Fork 0
mirror of https://github.com/LTTLabsOSS/markbench-tests.git synced 2026-01-08 05:33:52 -05:00
Code Issues Packages Projects Releases Wiki Activity

quick fixes and updates to implementation to maintain compatibility with CPU tests

Browse Source
This commit is contained in:
Arty Blue
2024-12-23 10:12:27 -08:00
parent e2c6599897
commit 104947c23d
2 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
procyon_ai/config/ai_computer_vision_tensorrt.def
View File

@@ -22,7 +22,7 @@
</setting>
<setting>
<name>ai_device_id</name>
<value>cuda:0</value> <!-- Get options from Procyon CLI: "ProcyonCmd.exe list-cuda-devices". Check "ProcyonCmd.exe -h" for correct syntax -->
<value></value> <!-- Get options from Procyon CLI: "ProcyonCmd.exe list-cuda-devices". Check "ProcyonCmd.exe -h" for correct syntax -->
</setting>
</settings>
</benchmark>

19
procyon_ai/ulprocai.py
View File

@@ -100,6 +100,8 @@ BENCHMARK_CONFIG = {
},
"Qualcomm_HTP": {
"config": f"\"{CONFIG_DIR}\\ai_computer_vision_snpe.def\"",
"device_id": "CPU",
"device_name": "CPU",
"process_name": "SNPE.exe",
"test_name": "Qualcomm SNPE (INTEGER)"
},
@@ -131,13 +133,16 @@ def get_arguments():
def create_procyon_command(test_option, process_name, device_id):
"""create command string"""
match process_name:
case 'WinML.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-winml-device {device_id}'
case 'OpenVino.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-openvino-device {device_id}'
case 'TensorRT.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-cuda-device {device_id}'
if device_id == 'CPU':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\"'
else:
match process_name:
case 'WinML.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-winml-device {device_id}'
case 'OpenVino.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-openvino-device {device_id}'
case 'TensorRT.exe':
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-cuda-device {device_id}'
command = command.rstrip()
return command