Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion QEfficient/base/modeling_qeff.py
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,11 @@ def _compile(

return self.qpc_path

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

if mdp_ts_json_path := compiler_options.pop("mdp_load_partition_config", None):
command.append(f"-mdp-load-partition-config={mdp_ts_json_path}")
Expand Down
2 changes: 1 addition & 1 deletion QEfficient/utils/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def get_models_dir():
ONNX_EXPORT_EXAMPLE_MIN_PS = 0.99
ONNX_EXPORT_OPSET = 13

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

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