o Y۷i:@s~dZddlmZddlZddlZddlZddlmZmZm Z e Z d&d d Z d'ddZ    d(d)ddZ  d*d+d$d%ZdS),zUtilities for creating high-performance keyword argument wrapper functions. This module provides tools for wrapping positional-only callables with keyword argument support using code generation techniques. ) annotationsN)AnyCallableIterablenames list[str]arg_typestrreturnNonecCstt|t|krtd|d||D]9}t|ts.t|dt|jd|t |r@td|d|d| sQtd|d|dqd S) aValidate that argument names are valid Python identifiers and unique. Parameters ---------- names List of argument names to validate. arg_type Description of the argument type (e.g., "Argument", "Keyword-only argument"). z Duplicate z names found in: z name must be a string, got z: zInvalid z name: z; is a Python keyword and cannot be used as a parameter namez! is not a valid Python identifierN) lenset ValueErrorlower isinstancer TypeErrortype__name__keyword iskeyword isidentifier)rrnamerR/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/utils/kwargs_wrapper.py_validate_argument_names"s"   r arg_names arg_defaultstuple kwonly_nameskwonly_defaultsdict[str, Any]reserved_namesset[str]c Cst|dt|tstdt|jt|t|kr+tdt|dt|dt|dt|}|D]}||vrFtd|d|q6t|}||@}|rXtd |||B} | |@} | rjtd | d d S) aValidate all input arguments for make_kwargs_wrapper. Parameters ---------- arg_names List of positional argument names. arg_defaults Tuple of default values for positional arguments. kwonly_names List of keyword-only argument names. kwonly_defaults Dictionary of default values for keyword-only arguments. reserved_names Set of reserved internal names that cannot be used as argument names. Argumentz"arg_defaults must be a tuple, got zarg_defaults has z values but only z positional argumentszKeyword-only argumentzDefault provided for 'z ' which is not in kwonly_names: z6Arguments cannot be both positional and keyword-only: z-Argument names conflict with internal names: z/. Please avoid using names starting with "__i_"N) rrrrrrr rr ) rrrrr!kwonly_names_setkey arg_names_setoverlapall_user_names conflictsrrr_validate_wrapper_argsAs8     r*r target_funcrlist[str] | Nonedict[str, Any] | None prototypeCallable | Nonecsn|durg}|dur i}d}dd|h}t||||||r/tt|t| d|ni}ggidfd d } |D]} | |vrR| | || qD| | qD|rd |D]} | |vrt| | || qf| | qfd} d} d| d|d| d} ||ti}i}t| |||d}|durtj ||dd|S)aCreate a wrapper with kwargs support for a function that only accepts positional arguments. This function dynamically generates a wrapper using code generation to minimize overhead. Parameters ---------- target_func The underlying function to be called by the wrapper. This function must only accept positional arguments. arg_names A list of ALL positional argument names in order. These define the positional parameters that the wrapper will accept. Must not overlap with kwonly_names. arg_defaults A tuple of default values for positional arguments, right-aligned to arg_names (matching Python's __defaults__ behavior). The length of this tuple determines how many trailing arguments have defaults. Example: (10, 20) with arg_names=['a', 'b', 'c', 'd'] means c=10, d=20. Empty tuple () means no defaults. kwonly_names A list of keyword-only argument names. These arguments can only be passed by name, not positionally, and appear after a '*' separator in the signature. Can include both required and optional keyword-only arguments. Must not overlap with arg_names. Example: ['debug', 'timeout'] creates wrapper(..., *, debug, timeout). kwonly_defaults Optional dictionary of default values for keyword-only arguments (matching Python's __kwdefaults__ behavior). Keys must be a subset of kwonly_names. Keyword-only arguments not in this dict are required. Example: {'timeout': 30} with kwonly_names=['debug', 'timeout'] means 'debug' is required and 'timeout' is optional. prototype Optional prototype function to copy metadata (__name__, __doc__, __module__, __qualname__, __annotations__) from. If None, no metadata is copied. Returns ------- A dynamically generated wrapper function with the specified signature Notes ----- The generated wrapper will directly embed default values for None and bool types and use a MISSING sentinel object to distinguish between explicitly passed arguments and those that should use default values for other types to ensure the generated code does not contain unexpected str repr. N__i_target_func __i_MISSING__i_arg_defaultsrr default_valuerr r c s|dur|d|dSt|tur0|rdnd}|d||dS|d||<d|d|dd | dS) zAAdd a parameter with a default value to arg_parts and call_parts.Nz=NoneTrueFalse=z["z"] if z is z else )appendrbool)rr3default_value_str_INTERNAL_DEFAULTS_DICT_INTERNAL_MISSING arg_parts call_partsruntime_defaultsrr_add_param_with_defaults   z4make_kwargs_wrapper.._add_param_with_default*z, z def wrapper(z): return (z) wrapperr)updated)rr r3rr r ) r*dictzipr r7joinMISSINGexec functoolsupdate_wrapper)r+rrrrr._INTERNAL_TARGET_FUNC_INTERNAL_NAMESarg_defaults_dictr@rarg_list call_listcode_str exec_globals namespacenew_funcrr:rmake_kwargs_wrappers\6 "         rU signatureinspect.Signatureexclude_arg_namesIterable[str] | Nonec Cs|durt|nt}g}g}g}i}d} |jD]b\} } | |vr#q| jtjjkr.td| jtjjkr9td| jtjj tjj fvrd| | | j tjj urYd} | | j q| rctd| dq| jtjjkr|| | | j tjj ur|| j || <qt|} t||| |||S)aCreate a wrapper with kwargs support for a function that only accepts positional arguments. This is a convenience function that extracts parameter information from a signature object and calls make_kwargs_wrapper with the appropriate arguments. Supports both required and optional keyword-only arguments. Parameters ---------- target_func The underlying function to be called by the wrapper. signature An inspect.Signature object describing the desired wrapper signature. prototype Optional prototype function to copy metadata (__name__, __doc__, __module__, __qualname__, __annotations__) from. If None, no metadata is copied. exclude_arg_names Optional iterable of argument names to ignore when extracting parameters from the signature. These arguments will not be included in the generated wrapper. If a name in this iterable does not exist in the signature, it is silently ignored. Returns ------- A dynamically generated wrapper function with the specified signature. Raises ------ ValueError If the signature contains *args or **kwargs. NFz)*args not supported in wrapper generationz,**kwargs not supported in wrapper generationTzRequired positional parameter 'z(' cannot follow parameters with defaults)r parametersitemskindinspect ParameterVAR_POSITIONALr VAR_KEYWORDPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDr7defaultempty KEYWORD_ONLYrrU) r+rVr.rXskip_setrarg_defaults_listrrhas_seen_positional_default param_nameparamrrrr"make_kwargs_wrapper_from_signature!sP%    rk)rrrr r r ) rrrrrrrr r!r"r r )rNNN)r+rrrrrrr,rr-r.r/r r)NN) r+rrVrWr.r/rXrYr r)__doc__ __future__rrJr]rtypingrrrobjectrHrr*rUrkrrrrs$   C $