U <ºc%oã"@s~UddlmZddlmZddlmZmZmZmZm Z m Z mZmZm Z ddlmZmZmZmZmZmZmZmZmZmZedƒZdddgZed d ƒZe eefZeddGd d„dƒƒZeddƒZeddƒZ eddƒZ!eddƒZ"eddƒZ#eddƒZ$eddƒZ%eddƒZ&eddƒZ'eddƒZ(eddƒZ)eddƒZ*eddƒZ+eddƒZ,edd ƒZ-edd!ƒZ.edd"ƒZ/edd#ƒZ0edd$ƒZ1edd%ƒZ2edd&ƒZ3edd'ƒZ4edd(ƒZ5edd)ƒZ6edd*ƒZ7edd+ƒZ8edd,ƒZ9edd-ƒZ:edd.ƒZ;edd/ƒZedd2ƒZ?edd3ƒZ@edd4ƒZAedd5ƒZBedd6ƒZCed7d8ƒZDedd9ƒZEedd:ƒZFedd;ƒZGedd<ƒZHedd=ƒZIejJeejKe ejLe!ejMe"ejNe#ejOe$ejPe&ejQe%ejRe'ejSe(ejTe)ejUe*ejVe+i ZWeeefeXd><ejYe#ejZe%ej[e*eje-ej\e.eje/ej]e0ej^e5ej_e7ej`e8ejae9ejbe:ejceejfe?ejgeFiZheeefeXd?<eddGd@dA„dAƒƒZieddGdBdC„dCƒƒZjeddGdDdE„dEƒƒZkeddGdFdG„dGƒƒZleddGdHdI„dIƒƒZmeddGdJdK„dKƒƒZneddGdLdM„dMƒƒZoeddGdNdO„dOƒƒZpeddGdPdQ„dQƒƒZqeddGdRdS„dSƒƒZre eielejekemenepeoeqerf ZseddGdTdU„dUƒƒZteddGdVdW„dWƒƒZueddGdXdY„dYƒƒZveddGdZd[„d[ƒƒZweddGd\d]„d]ƒƒZxeddGd^d_„d_ƒƒZyeddGd`da„daƒƒZzeddGdbdc„dcƒƒZ{eddGddde„deƒƒZ|eddGdfdg„dgƒƒZ}ddhœeeee didjœdkdl„Z~ddmlm€Z€mZm‚Z‚mƒZƒm„Z„m…Z…dnS)oé)Ú dataclass)ÚEnum) ÚDictÚIteratorÚListÚOptionalÚSequenceÚSetÚTupleÚTypeVarÚUnion) ÚArgumentÚBackendIndexÚBaseTyÚFunctionSchemaÚNativeFunctionÚNativeFunctionsGroupÚNativeFunctionsViewGroupÚ ScalarTypeÚSelfArgumentÚTensorOptionsArgumentsÚ_Tzat::TensorListz,const c10::List> &zconst at::ITensorListRef &ÚSpecialArgName)Z possibly_redundant_memory_formatT)Úfrozenc@s0eZdZUeeed<eed<edœdd„ZdS)ÚBaseCppTypeÚnsÚname©ÚreturncCs,|jdks|jdkr|jS|j›d|j›S)NÚz::)rr©Úself©r"ú6/tmp/pip-unpacked-wheel-gikjz4vx/torchgen/api/types.pyÚ__str__(szBaseCppType.__str__N)Ú__name__Ú __module__Ú__qualname__rÚstrÚ__annotations__r$r"r"r"r#r#s rrZuint8_tZint8_tZint16_tZint32_tZint64_tÚatÚHalfÚdoubleÚfloatZc10zcomplexzcomplexzcomplexÚboolÚBFloat16ÚvoidZstring_viewÚ GeneratorrÚTensorZOptionalTensorRefZ TensorListZITensorListRefZIOptTensorListRefÚDimnameZDimnameListÚ DimVectorÚLayoutÚDeviceÚScalarZOptionalScalarRefÚMemoryFormatÚQSchemeÚStorageÚStreamZIntArrayRefZOptionalIntArrayRefZOptionalSymIntArrayRefZ TensorOptionsztorch::autograd::generatedZTypeAndSizeZTensorGeometryÚSymIntZSymIntArrayRefÚscalar_tÚopmath_tÚScalarTypeToCppMappingÚBaseTypeToCppMappingc@sHeZdZUeed<ddœeedœdd„Zedœdd „Zd dœdd„Z d S)Ú BaseCTypeÚtypeF©Ú strip_ref©rDrcCs t|jƒS©N)r(rB©r!rDr"r"r#Úcpp_typeszBaseCType.cpp_typercCst|jƒ dd¡S)Nzat::r)r(rBÚreplacer r"r"r#Ú"cpp_type_registration_declarations”sz,BaseCType.cpp_type_registration_declarationsÚCTypecCs|SrFr"r r"r"r#Úremove_const_ref—szBaseCType.remove_const_refN) r%r&r'rr)r.r(rHrJrLr"r"r"r#rA‹s rAc@sHeZdZUded<ddœeedœdd„Zedœd d „Zddœdd„Zd S)Ú ConstRefCTyperKÚelemFrCrEcCs$|r|jj|dSd|j ¡›dS)NrCúconst ú &©rNrHrGr"r"r#rHŸszConstRefCType.cpp_typercCsd|j ¡›dS)NrOrP©rNrJr r"r"r#rJ¤sz0ConstRefCType.cpp_type_registration_declarationscCs |j ¡SrF©rNrLr r"r"r#rL§szConstRefCType.remove_const_refN© r%r&r'r)r.r(rHrJrLr"r"r"r#rM›s rMc@sHeZdZUded<ddœeedœdd„Zedœd d „Zddœdd„Zd S)ÚMutRefCTyperKrNFrCrEcCs"|r|jj|dS|j ¡›dS)NrCrPrQrGr"r"r#rH¯szMutRefCType.cpp_typercCs|j ¡›dS)NrPrRr r"r"r#rJ´sz.MutRefCType.cpp_type_registration_declarationscCs |j ¡SrFrSr r"r"r#rL·szMutRefCType.remove_const_refNrTr"r"r"r#rU«s rUc@sHeZdZUded<ddœeedœdd„Zedœd d „Zddœdd„Zd S)Ú OptionalCTyperKrNFrCrEcCsd|j ¡›dS©Nzc10::optional<ú>rQrGr"r"r#rH¿szOptionalCType.cpp_typercCsd|j ¡›dSrWrRr r"r"r#rJÃsz0OptionalCType.cpp_type_registration_declarationscCst|j ¡ƒSrF)rVrNrLr r"r"r#rLÆszOptionalCType.remove_const_refNrTr"r"r"r#rV»s rVc@sHeZdZUded<ddœeedœdd„Zedœd d „Zddœdd„Zd S)Ú ListCTyperKrNFrCrEcCsd|j ¡›dS©Nz c10::List sz'TupleCType.cpp_type..rX©ÚjoinrdrGr"r"r#rHszTupleCType.cpp_typercCsdd dd„|jDƒ¡›dS)NreracSsg|]}| ¡‘qSr")rJrfr"r"r#riszATupleCType.cpp_type_registration_declarations..rXrjr r"r"r#rJsz-TupleCType.cpp_type_registration_declarationscCstdd„|jDƒƒS)NcSsg|]}| ¡‘qSr")rLrfr"r"r#risz/TupleCType.remove_const_ref..)rcrdr r"r"r#rLszTupleCType.remove_const_refN) r%r&r'rr)r.r(rHrJrLr"r"r"r#rcs rcc@sHeZdZUeed<ddœeedœdd„Zedœdd „Zd dœdd„Z d S)ÚVectorizedCTyperNFrCrEcCsd|j ¡›dS)Nzat::vec::Vectorized.ú"c10::DispatchKeySet dispatchKeySetú, r~ú(ú)© rÚreturns_typer‡ÚreturnsrŠrHrŽrkr©r!rrr‘r›Zcpp_argsZcpp_args_strr"r"r#r€Ðsÿ zCppSignature.declcCsltj|jj|jd ¡}dd„| ¡Dƒ}|r8dg|}d |¡}|dkrV|| ¡}|›d|›d|›dS) Nr’cSsg|]}| ¡‘qSr"©r‚r”r"r"r#riîsz%CppSignature.defn..r–r—r~r˜r™ršrr"r"r#r‚äsÿ zCppSignature.defncCs<d dd„| ¡Dƒ¡}tj|jj|jd ¡›d|›dS)Nr—css|]}|jVqdSrF©rBr”r"r"r#Ú ÷sz(CppSignature.ptr_type..r’ú (*)(r™©rkrŽrr›r‡rœrŠrH©r!Zargs_types_strr"r"r#Úptr_typeöszCppSignature.ptr_typecCs<d dd„| ¡Dƒ¡}tj|jj|jd ¡›d|›dS)Nr—css|]}|jVqdSrFrŸr”r"r"r#r üsz$CppSignature.type..r’ú (r™r¢r£r"r"r#rBûszCppSignature.type)r%r&r'rr)r.r r(rŒrrsrŽrrr€r‚r¤rBr"r"r"r#r†Ÿs8 ûúûúr†c@sˆeZdZUeed<eed<eeed<eeed<eeed<edœdd„Zd d œee edœdd „Z eddœeeeddœdd„ƒZ dS)ÚCppSignatureGroupr‡Ú signatureÚfaithful_signatureÚsymint_signatureÚsymint_faithful_signaturercCs|jr|jS|jSdSrF)r¨r§r r"r"r#Úmost_faithful_signaturesz)CppSignatureGroup.most_faithful_signatureTr’©rŠrccs:|jV|jr|jV|r6|jr(|jV|jr6|jVdSrF)r§r¨r©rª)r!rŠr"r"r#Ú signaturesszCppSignatureGroup.signaturesF)rŒ)ÚfrˆrŒrcs‚ˆj‰tttdœ‡‡‡‡fdd„‰tttttfdœ‡‡fdd„}|dd\}}d}d}ˆ ¡rp|d d\}}tˆ||||d S)N)r‰rŠrcstˆ||ˆˆˆjdS)N)r‡r‰rŠrˆrŒr‹)r†r‹©r‰rŠ)r®rŒr‡rˆr"r#Úmake_sig"súz8CppSignatureGroup.from_native_function..make_sigr¬cs@d}ˆjjdk s tˆjjƒdkr,ˆd|d}ˆd|d}||fS)NrTr¯F)rŽZtensor_optionsÚlenÚout)rŠr¨r§)r‡r°r"r#Ú make_sigs,s z9CppSignatureGroup.from_native_function..make_sigsFr’T)r‡r§r¨r©rª)r‡r.r†r rÚ has_symintr¦)r®rˆrŒr³r§r¨r©rªr")r®rŒr‡r°rˆr#Úfrom_native_functions "ûz&CppSignatureGroup.from_native_functionN)r%r&r'rr)r†rr«r.rrÚstaticmethodrrµr"r"r"r#r¦s ÿþr¦c@sÞeZdZUeed<dZeed<dZeed<e e dœdd„Zedœd d „Zd e eedœd d„Zd!ddœe eeedœdd„Ze edœdd„Zedœdd„Zedœdd„Zedœdd„Zedddœeeeddœdd„ƒZdS)"ÚDispatcherSignaturer‡rrTrŠrcCstj|j|jdS©Nr’)Ú dispatcherrŽr‡rŠr r"r"r#rŽNszDispatcherSignature.argumentscCs|jt |j¡SrF)rr¹rr‡r r"r"r#rQszDispatcherSignature.nameNrocCsFd dd„| ¡Dƒ¡}|dkr(| ¡}| ¡ ¡›d|›d|›dS)Nr—css|]}| ¡VqdSrFr“r”r"r"r#r Usz+DispatcherSignature.decl..r~r˜r™)rkrŽrr›rH©r!rÚargs_strr"r"r#r€TszDispatcherSignature.declF)r‘)rr‘rcCsXdd„| ¡Dƒ}|r dg|}d |¡}|dkr:| ¡}| ¡ ¡›d|›d|›dS)NcSsg|]}| ¡‘qSr"ržr”r"r"r#ri]sz,DispatcherSignature.defn..r–r—r~r˜r™)rŽrkrr›rH)r!rr‘Úargsr»r"r"r#r‚Zs zDispatcherSignature.defncCsdd„| ¡DƒS)NcSsg|]}t|j|jƒ‘qSr")r„rrtr”r"r"r#rifsz-DispatcherSignature.exprs..)rŽr r"r"r#ÚexprseszDispatcherSignature.exprscCstj|jj|jdSr¸)r¹r›r‡rœrŠr r"r"r#r›hsz DispatcherSignature.returns_typecCs0d dd„| ¡Dƒ¡}| ¡ ¡›d|›dS)Nr—css|]}|jVqdSrFrŸr”r"r"r#r lsz/DispatcherSignature.ptr_type..r¡r™©rkrŽr›rH©r!Zdispatcher_args_types_strr"r"r#r¤kszDispatcherSignature.ptr_typecCs0d dd„| ¡Dƒ¡}| ¡ ¡›d|›dS)Nr—css|]}|jVqdSrFrŸr”r"r"r#r qsz+DispatcherSignature.type..r¥r™r¾r¿r"r"r#rBpszDispatcherSignature.type©rrŠ)r‡rrŠrcCst|||ƒSrF)r·)r‡rrŠr"r"r#Úfrom_schematszDispatcherSignature.from_schema)N)N)r%r&r'rr)rr(rŠr.rrsrŽrrr€r‚r„r½rKr›r¤rBr¶rÁr"r"r"r#r·Bs6 ÿÿþÿþr·c@s¤eZdZUeed<eed<dZeed<edœdd„Zde eed œd d„Z de eed œdd „Zedœdd„Ze edœdd„Zedœdd„Ze edœdd„ZdS)ÚNativeSignaturer‡rŠrrrcCs|jt |j¡SrF)rÚnativerr‡r r"r"r#r„szNativeSignature.nameNrocCsRd dd„| ¡Dƒ¡}|dkr(| ¡}tj|jj|jd ¡›d|›d|›dS)Nr—css|]}| ¡VqdSrFr“r”r"r"r#r ˆsz'NativeSignature.decl..r’r~r˜r™© rkrŽrrÃr›r‡rœrŠrHrºr"r"r#r€‡szNativeSignature.declcCsRd dd„| ¡Dƒ¡}|dkr(| ¡}tj|jj|jd ¡›d|›d|›dS)Nr—css|]}| ¡VqdSrFržr”r"r"r#r Žsz'NativeSignature.defn..r’r~r˜r™rÄrºr"r"r#r‚szNativeSignature.defncCs<d dd„| ¡Dƒ¡}tj|jj|jd ¡›d|›dS)Nr—css|]}| ¡VqdSrFržr”r"r"r#r •sz+NativeSignature.ptr_type..r’r¡r™)rkrŽrÃr›r‡rœrŠrH)r!r»r"r"r#r¤“szNativeSignature.ptr_typecCstj|j|jdSr¸)rÃrŽr‡rŠr r"r"r#rŽ˜szNativeSignature.argumentscCstj|jj|jdSr¸)rÃr›r‡rœrŠr r"r"r#r››szNativeSignature.returns_typecCstj| ¡t |j¡ddS)NF©rˆ)Ú translaterŽr¹r‡r r"r"r#Údispatcher_exprsžs ÿz NativeSignature.dispatcher_exprs)N)N)r%r&r'rr)r.rr(rrr€r‚r¤rrsrŽrKr›r„rÇr"r"r"r#rÂ{s rÂc@s2eZdZUeed<edœdd„Zedœdd„ZdS)ÚViewInverseSignatureÚgrcCs"|jjdk st‚tj|jdddS)NTF)Ú is_reverseÚinclude_namespace)rÉÚ view_copyÚAssertionErrorÚfunctionalizationrr r"r"r#r¨szViewInverseSignature.namecCsd|jjdk st‚t |jjj¡}dd„tj|jjjddDƒ}d| ¡›d| ¡›dd |¡›d S) NcSsg|]}| ¡‘qSr"r“r”r"r"r#ri¯sÿz-ViewInverseSignature.decl..T©rÊzstatic r~r˜r—ú);) rÉrÌrÍrÎr›r‡Úinner_argumentsrHrrk)r!Úreturn_typeÚdeclsr"r"r#r€¬sÿþzViewInverseSignature.declN)r%r&r'rr)r(rr€r"r"r"r#rÈ¤s rÈc@sneZdZUeed<eed<eedœdd„Ze dœdd„Z dd œeee d œdd„Ze eedd œdd„ƒZdS)ÚFunctionalizationLambdarÉrÊrcCsDt |jjj¡tjg}tj|jjj|jd}t j ||ddd}|S)NrÏFT)rˆZallow_expensive_conversions) r¹rŽrÉÚviewr‡rÎZreapply_views_bindingÚcapture_argumentsrÊrÆ)r!Z outer_ctxZcapture_bindingsZ capture_exprsr"r"r#Úcaptures¿sÿÿÿz FunctionalizationLambda.capturescCs`t |jjj¡}d dd„| ¡Dƒ¡}dd„tj|jdDƒ}d|›dd |¡›d | ¡›S) Nr—css"|]}|jj›d|j›VqdS)z = N)rBrr…)rgÚvalr"r"r#r Òsz/FunctionalizationLambda.decl..cSsg|]}| ¡‘qSr"r“r”r"r"r#riÕsÿz0FunctionalizationLambda.decl..rÏú[z](z) -> ) rÎr›rÉrÕr‡rkr×Úouter_argumentsrÊrH)r!rÒZcapture_strrÓr"r"r#r€Ðs ÿþzFunctionalizationLambda.declN)Ú reapply_views)rÛrc CsØtj|j|jd|d}tj|jd}tj|jjj|jd}||}|jjdk sRt ‚tj |jjj|jd}t |jjj¡}dd„tj||ddDƒ}|js¾|dk r¾|›dd |¡›d |j›dS|›dd |¡›dSdS) NT)rÊrËrÛrÏcSsg|] }|j‘qSr")r…rfr"r"r#riîsz6FunctionalizationLambda.inner_call..FrÅr˜r—z)[z];rÐ)rÎrrÉrÊrÚrÖrÕr‡rÌrÍrÑZinner_call_indexrÆrk) r!rÛZinner_call_nameZarg_ctxZcapture_ctxZfull_ctxZ call_bindingsZmaybe_indexZ call_exprsr"r"r#Ú inner_callÛs0üÿÿÿz"FunctionalizationLambda.inner_call)rÉrÊrcCs t||ƒSrF)rÔ)rÉrÊr"r"r#Ú from_funcösz!FunctionalizationLambda.from_func)r%r&r'rr)r.rr„r×r(r€rrÜr¶rÝr"r"r"r#rÔ¸s þrÔc@sFeZdZUeed<eed<d eeedœdd„Zee dœdd „Z dS)ÚStructuredImplSignaturerÉrNrocCs,d dd„| ¡Dƒ¡}d|j›d|›dS)Nr—css|]}| ¡VqdSrFržr”r"r"r#r sz/StructuredImplSignature.defn..zTORCH_IMPL_FUNC(z)(r™)rkrŽrrºr"r"r#r‚szStructuredImplSignature.defnrcCst |j¡SrF)Ú structuredZimpl_argumentsrÉr r"r"r#rŽsz!StructuredImplSignature.arguments)N)r%r&r'rr)r(rr‚rrsrŽr"r"r"r#rÞýs rÞ)r)rÂr·)r®Ú backend_indexrrcCsf| |¡}|dk o| ¡}|r:|j ¡s:td|j›dƒ‚|jrRtj|j||dSt |j||dSdS)Nz&attempted to define symint kernel for z without SymInt in schemarÀ) Z get_kernelZsupports_symintr‡r´rÍZdispatch_keyZexternalr·rÁrÂ)r®ràrÚmetarŠr"r"r#Úkernel_signature s ÿþrâ)rr¹rÎrÃrßrÆN)†ZdataclassesrÚenumrÚtypingrrrrrr r rrZtorchgen.modelr rrrrrrrrrrZTENSOR_LIST_LIKE_CTYPESrr(rrrZbyteTZcharTZshortTZint32TZlongTZhalfTZdoubleTZfloatTZcomplexHalfTZ complexFloatTZcomplexDoubleTZboolTZ bfloat16TZvoidTZstringTZ generatorTZscalarTypeTZtensorTZoptionalTensorRefTZtensorListTZiTensorListRefTZiOptTensorListRefTZdimnameTZdimnameListTZ dimVectorTZlayoutTZdeviceTZscalarTZoptionalScalarRefTZ memoryFormatTZqschemeTZstorageTZstreamTZintArrayRefTZoptionalIntArrayRefTZoptionalSymIntArrayRefTZtensorOptionsTZtypeAndSizeTZtensorGeometryTZSymIntTZsymIntArrayRefTr=r>ZByteÚCharZShortZIntZLongr+ZFloatÚDoubleZComplexHalfZComplexFloatZ ComplexDoubleZBoolr/r?r)rbr-r.r1r2r3r4r5r6r7r8r9r:r;r<r@rArMrUrVrYr[r\r^rcrlrKrnrsr„r†r¦r·rÂrÈrÔrÞrâZtorchgen.apirr¹rÎrÃrßrÆr"r"r"r#Úsl,0 ý ÿ óï÷ÿ< d=8(Dÿþ