@@ -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