o 0 h@sddlmZddlmZddlmZmZddlmZm Z m Z m Z m Z m Z mZer8ddlmZddlmZmZmZdZdd d ZeddGdddZdS)) annotations) dataclass)Callable TYPE_CHECKING)ArgumentBaseTyBaseTypeListTypeNativeFunction OptionalTypeType)Sequence)BindingCType NamedCTypez fr returnstrcCs |jjS)N)funcnameunambiguous_name)rrT/var/www/vscode/kcb/lib/python3.10/site-packages/torchgen/executorch/api/unboxing.pyrs rT)frozenc@sReZdZUdZded<d"dd Zd d d#ddZd$ddZd%ddZd&dd Z d!S)'Unboxinga Takes a sequence of Bindings and unbox EValues to these Bindings. Return generated code that performs correct unboxing. A sample generated code: // aten::mul.out(Tensor self, Tensor other, *, Tensor(a!) out) -> Tensor(a!) void mul_out(EValue** stack) { EValue& self = *stack[0]; EValue& other = *stack[1]; EValue& out = *stack[2]; const torch::executor::Tensor & self_base = self.to(); const torch::executor::Tensor & other_base = other.to(); torch::executor::Tensor & out_base = out.to(); EXECUTORCH_SCOPE_PROF("native_call_mul.out"); torch::executor::mul_outf(self_base, other_base, out_base); } zCallable[..., NamedCType]argument_type_genargsSequence[Binding]rtuple[list[Binding], list[str]]c sfddttD}g}D]3}t|jts td||j}|j|j|j|j d\}}}} | | | || | |q||fS)Ncs$g|]}d|jd|dqS)zEValue& z = *stack[z];)r).0irrr ?s$z.Unboxing.convert_arguments..z7Unexpected argument type, expecting `Argument` but got mutable) rangelen isinstanceargumentr Exceptionargumenttype_evalue_converttyperis_writeextendappend with_name) selfr code_list binding_listargr( unboxed_name_codedeclrr!rconvert_arguments<s    zUnboxing.convert_argumentsFr#tr arg_namerr$bool'tuple[str, CType, list[str], list[str]]cCs|j|||dj}t|tr|d}|j|||d\}}n6t|tr4|d}|j||||d\}}n t|trJ|d}|j||||d\}}n t d|d|||||fS) a Takes in the type, name and mutability corresponding to an argument, and generates a tuple of: (1) the C++ code necessary to unbox the argument (2) A Binding corresponding to the newly created unboxed variable, including variable name and its CType :param t: a `Type` of an argument :param arg_name: argument name :param mutable: boolean for whether this argument type is mutable :return: unboxed result )r$binds_base)r:out_namectype_opt_out)r:r?r9r@ _list_outzCannot handle type z . arg_name: ) rr+r'r_gen_code_base_typer _gen_code_optional_typer _gen_code_list_typer))r0r9r:r$r@r?r6r7rrrr*Ps(        z$Unboxing.argumenttype_evalue_convertr?r@rtuple[list[str], list[str]]c Cs.|d|d|d|jdddggfS)N  = z.to();)cpp_type)r0r:r?r@rrrrCss&zUnboxing._gen_code_base_typer c CsJ|d}||j|\}}}} d|d|d|jdddd| fS) N_opt_in auto rHz .toOptional();  )r*elemrKsplit) r0r:r?r9r@in_nameres_name base_typeres_coder7rrrrDzs   z Unboxing._gen_code_optional_typer c Csh|d}|d}g}||j|\}} } } t|jtr7|jjtjkr7|d|d|dd|| fSt|jtr^|jjtj ksK|jjtj kr^|d|d|dd|| fSt|jtr~|jjtj kr~|d|d|dd|| fSt|jtr|jjtj kr|d |j d |d |d|d |d |d|d|d|d|dd|| fSt|jtrt|jjtr|jjjtjkr|d|d|d|d|d |d|d|d|d|dd|| fS|d} | d| jddd | d|d|d |dt| d| d|d |jddd!|d"| d#d|| fS)$N_list_in_elemrMrHz!.toTensorList(); rNz.toIntList(); z!.toDoubleList(); z& #ifdef USE_ATEN_LIB std::array z; auto z(.toBoolList(); size_t _i = 0; for (auto z: z) { z [_i++] = z; } #else auto z&.toBoolList(); #endif z #ifdef USE_ATEN_LIB auto z@.toListOptionalTensor(); c10::List<::std::optional> z ; for (auto z .push_back(z); } #else auto z0.toListOptionalTensor(); #endif _vecz std::vectors $