Skip to content

Commit 244fa42

Browse files
committed
adding aic-hw-version compile options
Signed-off-by: Abukhoyer Shaik <[email protected]>
1 parent 7e0ad94 commit 244fa42

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

QEfficient/base/modeling_qeff.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,11 @@ def _compile(
276276

277277
return self.qpc_path
278278

279-
command = constants.COMPILER + [f"-m={onnx_path}"]
279+
command = (
280+
constants.COMPILER
281+
+ [f"-aic-hw-version={compiler_options.pop('aic-hw-version', '2.0')}"]
282+
+ [f"-m={onnx_path}"]
283+
)
280284

281285
if mdp_ts_json_path := compiler_options.pop("mdp_load_partition_config", None):
282286
command.append(f"-mdp-load-partition-config={mdp_ts_json_path}")

QEfficient/utils/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def get_models_dir():
7070
ONNX_EXPORT_EXAMPLE_MIN_PS = 0.99
7171
ONNX_EXPORT_OPSET = 13
7272

73-
COMPILER = ["/opt/qti-aic/exec/qaic-exec", "-aic-hw", "-aic-hw-version=2.0"]
73+
COMPILER = ["/opt/qti-aic/exec/qaic-exec", "-aic-hw"]
7474

7575
# InternVL constants
7676
# Fixing the feature size with reference to OpenGVLab/InternVL2_5-1B, OpenGVLab/InternVL2_5-38B and OpenGVLab/InternVL2_5-78B

0 commit comments

Comments
 (0)