U <c-@sddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZeddGd d d ZeddGd d d ZeddGd ddeZeddGdddZ eddGddde Z!eddGdddZ"eddGdddZ#eddGdddZ$eddGdddZ%eddGdddZ&ddee'edd d!Z(ee'd"d#d$Z)ddd%ee'e'e*d&d'd(Z+eee,d)d*d+Z-eed,d-d.Z.ddd/ee'e'e d0d1d2Z/ddd/eee*e'e'e d3d4d5Z0eed6fee*d7d8d9Z1ee*d)d:d;Z2ee*d)dd?d@Z4e e*d>dAdBZ5dte ee'ee$d6fdCdDdEZ6dFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZhZ7ee*d"d[d\Z8ee*d"d]d^Z9d_d`eee ee*d6fdadbdcZ:dddeee*ee*e'e*dedfdgZ;ddde,ee'e%dhdidjZddde ee'e&dCdrdsZ?d_S)u) dataclass)DictListOptionalSequenceSetTupleUnion)cpp)Binding CppSignatureCppSignatureGroup)pythonify_default) ArgumentBaseTyBaseTypeFunctionSchemaListTypeNativeFunction OptionalTypeReturnTypeVariantT)frozenc@seZdZUeedfed<dS) PythonReturns.returnsN)__name__ __module__ __qualname__rr__annotations__r r 7/tmp/pip-unpacked-wheel-gikjz4vx/torchgen/api/python.pyrs rc@sjeZdZUeed<eed<eeed<eeed<dddeeedd d Zddd eeed d dZ dS)PythonArgumentnametypedefault default_initFTmethodsymint)r(r)returncCst|j|ddddd}|j}|dkr<|dkr<|seZdZUeedfed<eeedfeddddZdS)rJ.outputs)rRr*cCs|sdSt|}|dkr6t|dj|djdd|dS|dkr|ttdd|r^td|tdttt j |dd|dSt d dS) NrErr0)r#r$r%r&rRcSs |j SNr$is_tensor_likear r r!>z0PythonOutArgument.from_outputs..zUnsupported output type: r?z!Unexpected PythonOutArgument size) lenrJr#r$anymap RuntimeErrorrrrr.AssertionError)rRsizer r r! from_outputs-s,zPythonOutArgument.from_outputsN) rrrrr"r staticmethodrr`r r r r!rJ$s  rJc@seZdZUeed<eedfed<eedfed<eeed<e ed<eedfed<e ed<e e d d d Z d d d e e ee eefdfdddZed ddZed ddZd dde e edddZd de edddZd de eedddZd S)!PythonSignaturer#. input_args input_kwargs output_argsrtensor_options_argsr(r*cCsdS)NFr r-r r r!r<hszPythonSignature.deprecatedF) skip_outputsskip_tensor_options)rirjr*cCsNg}||j||j|jdk r6|s6||j|sF||jt|SrS)extendrcrdreappendrftuple)r-rirjresultr r r! argumentsls    zPythonSignature.argumentscCs t|SrS)rZrorhr r r!arguments_countxszPythonSignature.arguments_countcCst|jt|jSrS)rZrcrdrhr r r! output_idx{szPythonSignature.output_idxTrir)rir)r*cs^j|d}ttfdd|}tj}t||krF||djdd|dS)Nrics|jjdS)Nr')r:r(rVr-r)r r!rXrYz/PythonSignature.signature_str..*rD, rG)rolistr\rZrcinsertr#join)r-rir)argsschema_formalspositional_argcr rur! signature_strs    zPythonSignature.signature_strrtrir*cs~j|d}ttfdd|}tj}t||krD||dt}jr^|dddjdd |d |d S) Nrtcs|jjdSNr(rOr(rVrhr r!rXrYz3PythonSignature.signature_str_pyi..rvrr-def rDrw) -> : ...) rorxr\rZrcryreturns_str_pyir(r#rzr-rir{r|r} returns_strr rhr!signature_str_pyis     z!PythonSignature.signature_str_pyic sj|d}ttfdd|}}tj}d}|dkrn|dj}t|trnt |j dkrn|dkrnd}|svdSd |dj d |d<t }j r|dd d j d d|d|dS)Nrtcs|jjdSrrrVrhr r!rXrYz:PythonSignature.signature_str_pyi_vararg..Fr)rLSymIntrETrvz: _intr-rrDrwrr)rorxr\rprZrcr$rIrrPrKr#rr(ryrz) r-rir{r|num_argsZnum_positionalargsZhave_vararg_versionZ vararg_typerr rhr!signature_str_pyi_varargs.     z(PythonSignature.signature_str_pyi_varargN)rrrrPrrr"rrJrrQpropertyr<r rorLrprqr~rrr r r r!rbKs*     rbc@seZdZUeed<eedfed<eedddZ ddd eeed d d Z dd eedddZ dd ee edddZ dS)PythonSignatureDeprecateddeprecated_schema.deprecated_args_exprsrgcCsdS)NTr rhr r r!r<sz$PythonSignatureDeprecated.deprecatedFTrrrscCstj|||ddS)Nrrz |deprecated)rbr~)r-rir)r r r!r~sz'PythonSignatureDeprecated.signature_strrtrcslj|d}ttfdd|}tj}t||krD||dt}djdd|d|d S) Nrtcs|jjddS)NTr;rrVrhr r!rXrYz=PythonSignatureDeprecated.signature_str_pyi..rvrrDrwrr) rorxr\rZrcryrr#rzrr rhr!rs    z+PythonSignatureDeprecated.signature_str_pyicCsdSrSr )r-rir r r!rsz2PythonSignatureDeprecated.signature_str_pyi_varargN)rrrrrrrPrrQr<r~rrrr r r r!rs   rc@seZdZUeed<eed<dS)!PythonSignatureNativeFunctionPair signaturefunctionN)rrrrbrrr r r r!rs rc@sDeZdZUeed<eed<eeed<eeeeddddZ dS)PythonSignatureGrouprbaseoutplace) functionalr?r*cCsP|dkrt|j|jddS|jj}|jj|d<tt|jf||j|jdS)N)rrrrf)rrr__dict__copyrfr$)clsrr?Zsignature_kwargsr r r! from_pairss  zPythonSignatureGroup.from_pairsN) rrrrbrrr classmethodrrr r r r!rs  rc@s&eZdZUeed<eed<eed<dS)DispatchLambdaArgumentr#r9 is_out_argN)rrrrPrrQr r r r!r2s rc@s@eZdZUeed<eed<eed<eed<eedddZdS) PythonArgParserOutputExprr#exprindexargumentrgcCsd|jdS)Nz _r.isNone(rG)rrhr r r! is_none_exprMsz&PythonArgParserOutputExpr.is_none_exprN) rrrrPrrLr"rrr r r r!r>s rc@s&eZdZUeeed<eeed<dS)DispatchLambdaArgumentExprsexprsinitsN)rrrrrPrr r r r!rWs rFr)fr(r*cCstj||djSr)r Zfrom_native_functionr)rr(r r r!_cpp_signatureqsr)rr*cCs|jjjdk SrS)funcrotensor_optionsrr r r!has_tensor_optionsusr simple_typer))trr)r*cCsXt|tr|jtjkrdS|jtjkr*dS|jtjkr:dS|jtjkrJdS|jtjtj tj tj tj tj tjtjtjtjtjtjtjf kr|jjSnt|trt|jdkrdSt|j||d}|dSt|trB|s|jnd}t|jdkr|jdk std |jd St|jd kr<|dk r8d |d SdSt|jdkr|rl|dk rhd|d SdS|dk rd |d SdSnt|jdkr|dk rd|d SdSt|jdkr|dk rd|d SdSt|jdkr|rdSdSn*t|jdkr&|dk r"d|d SdSt|j||d}d|d Stdt|dS)Nr.int64_tdoublezc10::string_viewTensor?r?rQz::std::arrayrLz IntArrayRef[rAZ IntArrayRefrzSymIntArrayRef[ZSymIntArrayRefz TensorList[Z TensorListScalarz ScalarList[Z ScalarListz c10::List>z(const c10::List> &Dimnamez DimnameList[Z DimnameListz ArrayRef<unrecognized type )rIrr#rr.rLfloatrPrQQSchemer ScalarType GeneratorStorageLayoutDevice MemoryFormatrStreamZConstQuantizerPtrrrrKr6rr_r^r]repr)rrr)rKr_r r r!r6sn          r6)rr*cCs,|}|dk r$t|jdkr$|jSdSdS)NrQ)Z is_list_likerPrKr_)rlr r r!argument_type_sizesr)rWr*c Cs6t|j|j|jdk r,ttt|j|jndddS)Nr#r$r%r&)r"r#r$r%rPrr default_exprrVr r r!rsr)r(pyi)rr(rr*cCst|j|j||dS)N)category_overrider(r)signature_from_schemarr)rr(rr r r!rs r)rrr(rr*c sg}|jj|s2jjdk r2|jjj|jj|jj|jj|jj t ddjj Dt ddjj Dt ddjj Dt tttfdd|}t tttfdd|}t tttfdd|}td djjD}td dDr6td td djD} t} |d kph| oh| } |dkp| dp| d} g} | s| rtttdfdd }| tdtttjd| rdn|dd| tdtttj d| rdn|dd| tdtttj!d| r.dn |dp:dd| tdtttj"ddd| tdtttj"dddt#jd }t$tjj||t%&|t | ||d!S)"Ncss|] }|jVqdSrSr#.0rWr r r! sz(signature_from_schema..css|] }|jVqdSrSrrr r r!rscss|] }|jVqdSrSrrr r r!rscs |jkSrSrrV) input_arg_setr r!rXrYz'signature_from_schema..cs |jkSrSrrV)kwarg_only_setr r!rXrYcs |jkSrSrrV) out_arg_setr r!rXrYcss|]}|jVqdSrSrTrr r r!rscss|]}|jdkVqdS) requires_gradNrrr r r!r szTargument named requires_grad is reserved, should not explicitly add it in the schemacss|]}|jVqdSrSrTrrr r r!rsfactory)newlikeZnew__like)r#r*csFjj}|dkrdSt||}|jdks2|jdkr6dSt|j|jS)Nr0)rorgetattrr%r rr$)r#Z topt_argsrW)rr r!topt_default_init s z0signature_from_schema..topt_default_initdtyper0zself.scalar_type()rlayoutz self.layout()devicez self.device()z$torch::tensors::get_default_device() pin_memoryFalserr)r#rcrdrerfrr()'rkroZpre_self_positionalZself_argrlrZpost_self_positionalZpre_tensor_options_kwarg_onlyZpost_tensor_options_kwarg_onlyr?setZflat_positionalZflat_kwarg_onlyrmr\filterr[Z flat_non_outZschema_order_arguments ValueErrorrr r#rMrNrPrr"rrrrrrrQrrbrJr`)rrr(rr{rcrdrRZhas_tensor_input_argZhas_tensor_returnr#Zis_factory_functionZis_like_or_new_functionrfrrr )rrrrr!rs                 r.)rr*cCsRt|dksttdd|r"gSttdd|r.cSs |jdkSrSrrr r r!rXyrYz)Unnamed field is not supported by codegencSs t|jSrS)rPr#rr r r!rXrY)rZallr\r[rrxrr r r!namedtuple_fieldnamesus  rcCsd}t|tr|j}d}t|tr |jtjkr4d}|jtjkrFd}n|jtjkrXd}n|jtj krjd}n|jtj kr|d}n|jtj krd}n|jtj krd }n~|jtj krd }nl|jtjkrd }nZ|jtjkrd }nH|jtjkrd }n6|jtjkrd}n$|jtjtjtjtjfkr|jj}nt|trt |jdkrR|jdk rLdnd}n^|rt|jtrnd}|jdk r~dnd}n,t |jdkrd}nt|j}d|d}ntdt||rd|d}|S)NFT_intzUnion[_int, SymInt]Z_floatrPNumberZ_dtypeZ_boolZ_qschemeZ_layoutzUnion[_device, str, None]Z memory_formatzUnion[str, ellipsis, None]rLzUnion[_int, _size]_sizez/Union[Tensor, Tuple[Tensor, ...], List[Tensor]]z'Union[Tuple[Tensor, ...], List[Tensor]]rzSequence[_float]z Sequence[rArr@)rIrrKrr#rrLrrrPrrrQrrrrrr.rrrrr_rUrHr]r)rZ add_optionalretrKr r r!rHsb                   rHcCs|t|tr t|j}d|dSt|trT|jtjkr:dS|jtjkrLd}nt |St|t rtt|j}d|dSt |S)Nr@rAZ_devicez Optional[str]zList[) rIrreturn_type_str_pyirKrr#rrrrHr)rinnerrr r r!rs         r)rr*cCs`dd|jjD}|j}t|jj}|r\ddt||D}d|dd|d}||fSdS)NcSsg|]}t|jqSr rr$rr r r! sz+returns_named_tuple_pyi..cSs"g|]\}}d|d|dqS)z("z", rGr )rr#typr r r!rsz NamedTuple("z", [rwz]))rr#rziprz)rpython_returnsZnamedtuple_name field_namesZ tuple_argsZnamedtuple_defr r r!returns_named_tuple_pyis rcCsdt|jj}|rd|jSdd|jjD}t|dkrLdd|dSt|dkr`|dSd S) Nztorch.return_types.cSsg|]}t|jqSr rrr r r!rsz#returns_str_pyi..rEzTuple[rwrArr0)rrr#rZrz)rrrr r r!rs    r)psrr)r*csjttrj}n|j}tj|jd|d|jd}tdd|jjDt t dfdd }t t ||S)NF)roZfaithfulr)r(cpp_no_default_argscss|] }|jVqdSrSrrr r r!r1sz'dispatch_lambda_args..)cpp_argr*cs\|j}|jk}jr&|jdkr&d}n&tdkp6| }|rLddi||}t|j||dS)Nr-zconst at::Tensor &rEz at::Tensor & at::Tensor)r#r9r)r$r#r(rZr8r)rr9rZensure_temp_safeZout_argsrr r!dispatch_lambda_arg4s" z1dispatch_lambda_args..dispatch_lambda_arg) rIrrrr rorrr?r rrmr\)rrr)Zschemacpp_argsrr rr!dispatch_lambda_args!s rrz#::std::tuplez.::std::tuplez9::std::tuplezD::std::tuplez6::std::tuplez2::std::tuplezA::std::tuplez=::std::tuplez::std::tuplez2::std::tuple>z::std::vectorz at::ScalarrQrzvoid*voidz at::QSchemerzat::IntArrayRefzat::ScalarTypecCsJttdd|jj}tj|dd}|tkrFt|jj d||S)NcSst|j|jdSrS)rr#r$rr r r!rX~rYz,dispatch_lambda_return_str..Tr+z returns unsupported type ) rmr\rrr Z returns_typeZcpp_typeSUPPORTED_RETURN_TYPESr]r#)rZreturns_without_annotationZ return_strr r r!dispatch_lambda_return_strisrcCs|j}tj|j|d}tj|jkr0d|Stj|jkrnt|sV|jjjj dr\d}nd}|d|St d|jdS)N)Zsymint_overloadzself.rZtorchatz::z1could not dispatch, neither function nor method: ) rZ has_symintr r#rr(variantsrrrrNr])rr)r# namespacer r r!cpp_dispatch_targets    rN)python_signature)rrr*csltdd}t}t|ts4ttdd|}nttfdd|j}tj j krhttdj |}|S)NFrcSs|jSrSrrVr r r!rXrYz$cpp_dispatch_exprs..cs|dkpjS)Nr?)r is_out_fn)nrr r!rXrYr-) rrormrIrr\rrrr(r__ne__)rrrrr rr!cpp_dispatch_exprss   rr+)rr%r&r)r*cCsv|dk }|r(t|dkr(td|dt|tr$|jtjtjtjtj tj fkr^|jj S|jtj krv|rrdSdS|jtj kr|rdSdS|jtjkrdS|jtjkr|rd SdSnh|jtjkr|rd Sd S|jtjkrd S|jtjkrd S|jtjkr|r dSdS|jtjkrbdSn>t|trt|jdkrDdSt|jdkrXdSt|jdkrldS|s|dkrt|jdd|ddSt|j|||dSnt|trbt|jdkr|jdk rd|jdSdSnt|jdkrdSt|jdkr d St|jd!krd"St|d#kr0d$St|jd%krP|rJd&Sd"Snt|d'krbd(Std|d)dS)*N) ScalarType?rrDevice?rLayout?rQbool?ztype 'z+' does not supported unpacking with defaultZscalartypeWithDefaultZ scalartypeZdeviceWithDefaultrZtoInt64ZtoSymIntZtoBoolWithDefaultZtoBoolZtoDoubleZ stringViewZlayoutWithDefaultrZ memoryformatr.ZoptionalTensorr generatorz Dimname[]ZtoDimnameListOptional)Nr0r2r+rz tensorlist_n1s z+arg_parser_output_exprs..) enumeratero)rrr)r r+r!arg_parser_output_exprs.s  rrrrr )rrrrrc sJt|||d}t|||d}g}it|}|jddD]}|j}||jj} |rx|dkrx|d| dg||<q8t|trt |j dkr|j r|d| dgt |j D]\} } d | d | j<qq8t|jd kr |d |d | dd|d|d|dg||<q8| |<q8|jr:dd<ttdd|j|r^|j rpt|j d|jD]f}|jtkrt|j d|jdt|jt|jkrvt|j dt|jd|jdqvttfddtst|j d|d|djd|djd |d!jd"|d#jd$|d%jd& d'd'<|s,r,dkr |j st|j d(ttfd)dd*st|j d+|d,|d-jd.|djd/|djd.|d!jd/|djd.|djd0 d#kr,t|j d1d tttfd2d||d3S)4Nr+T)rjr-z auto self = ;rEz auto out = zout[rAz Dimname[]?zauto __z = zc10::optional z = __z$ ? c10::make_optional(DimnameList(__z.value())) : c10::nullopt;cSs|jSrSrrVr r r!rX~rYz'dispatch_lambda_exprs..z : tensor options with output argz%: unrecognized tensor options field 'z' in python binding argumentsz: unrecognized type 'z' for tensor options field ''cs|kSrSr rVtensor_options_args_namesr r!rXrYz": incomplete tensor options args: z0const auto options = TensorOptions() .dtype(rz) .device(rz) .layout(rz) .requires_grad(rz) .pinned_memory(rz1); torch::utils::maybe_initialize_cuda(options); optionsz1: dtype in tensor_options_args without output argcs|kSrSr rVrr r!rXrY)rrz,: incomplete tensor options for output checkzcheck_out_type_matches(r?rwz, z); zE: expected "requires_grad" in tensor_options_args absent, but found [cs |jSrSrrV)lambda_args_exprsr r!rXrY)rr)rrrror#rrkrIrJrZrRrrrrPr$r(rxr\rfr]TENSOR_OPTIONS_FIELDSr8rkeysrlrrrm) rrr)Zarg_parser_outputsZ lambda_argsrZ has_toptionsrWr#Zarg_parser_exprrZout_argr )rrr!dispatch_lambda_exprsBs                  r)T)@Z dataclassesrtypingrrrrrrr Z torchgen.apir Ztorchgen.api.typesr r r Z torchgen.genrZtorchgen.modelrrrrrrrrrrrr"rJrbrrrrrrrQrrrPr6rLrrrrrrHrrrrrrrrr rrrrr r r r!s  $  0?X&y3/  C  @ 6 0 ( a