Skip to content

Commit 767f7fe

Browse files
author
Amit Raj
committed
Hash fix
Signed-off-by: Amit Raj <[email protected]>
1 parent e822679 commit 767f7fe

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

QEfficient/utils/export_utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,11 +101,13 @@ def _generate_export_hash(qeff_model, args, kwargs, func):
101101
Returns:
102102
Tuple of (export_hash: str, filtered_hash_params: dict)
103103
"""
104+
# Extract use_onnx_subfunctions before binding (it's used by wrapper, not _export)
105+
use_onnx_subfunctions = kwargs.pop("use_onnx_subfunctions", False)
106+
104107
# Extract function signature
105108
original_sig = inspect.signature(func)
106109
params = list(original_sig.parameters.values())[1:] # Skip 'self'
107110
new_sig = inspect.Signature(params)
108-
109111
# Bind all arguments
110112
bound_args = new_sig.bind(*args, **kwargs)
111113
bound_args.apply_defaults()
@@ -130,7 +132,7 @@ def _generate_export_hash(qeff_model, args, kwargs, func):
130132
dynamic_axes=all_args.get("dynamic_axes"),
131133
export_kwargs=all_args.get("export_kwargs", None),
132134
onnx_transform_kwargs=all_args.get("onnx_transform_kwargs", None),
133-
use_onnx_subfunctions=all_args.get("use_onnx_subfunctions", False),
135+
use_onnx_subfunctions=use_onnx_subfunctions,
134136
)
135137

136138
return export_hash, filtered_hash_params

0 commit comments

Comments
 (0)