mirror of
https://github.com/LTTLabsOSS/markbench-tests.git
synced 2026-01-08 21:48:00 -05:00
quick fixes and updates to implementation to maintain compatibility with CPU tests
This commit is contained in:
@@ -22,7 +22,7 @@
|
|||||||
</setting>
|
</setting>
|
||||||
<setting>
|
<setting>
|
||||||
<name>ai_device_id</name>
|
<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>
|
</setting>
|
||||||
</settings>
|
</settings>
|
||||||
</benchmark>
|
</benchmark>
|
||||||
@@ -100,6 +100,8 @@ BENCHMARK_CONFIG = {
|
|||||||
},
|
},
|
||||||
"Qualcomm_HTP": {
|
"Qualcomm_HTP": {
|
||||||
"config": f"\"{CONFIG_DIR}\\ai_computer_vision_snpe.def\"",
|
"config": f"\"{CONFIG_DIR}\\ai_computer_vision_snpe.def\"",
|
||||||
|
"device_id": "CPU",
|
||||||
|
"device_name": "CPU",
|
||||||
"process_name": "SNPE.exe",
|
"process_name": "SNPE.exe",
|
||||||
"test_name": "Qualcomm SNPE (INTEGER)"
|
"test_name": "Qualcomm SNPE (INTEGER)"
|
||||||
},
|
},
|
||||||
@@ -131,13 +133,16 @@ def get_arguments():
|
|||||||
|
|
||||||
def create_procyon_command(test_option, process_name, device_id):
|
def create_procyon_command(test_option, process_name, device_id):
|
||||||
"""create command string"""
|
"""create command string"""
|
||||||
match process_name:
|
if device_id == 'CPU':
|
||||||
case 'WinML.exe':
|
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\"'
|
||||||
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-winml-device {device_id}'
|
else:
|
||||||
case 'OpenVino.exe':
|
match process_name:
|
||||||
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-openvino-device {device_id}'
|
case 'WinML.exe':
|
||||||
case 'TensorRT.exe':
|
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-winml-device {device_id}'
|
||||||
command = f'\"{ABS_EXECUTABLE_PATH}\" --definition={test_option} --export=\"{REPORT_PATH}\" --select-cuda-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()
|
command = command.rstrip()
|
||||||
return command
|
return command
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user