JThvR %SSKJr SSKrSSKrSSKrSSKJr SSKJrJr SSK J r J r J r SSK JrJrJr \ (aSSKJrJr \"SS 9"S S 55r"S S \5rSrSR/5r/SQrS/\Vs/sHnSU-PM sn-rSS1r"SS\5r"SS\5rSfSjr\H<r\H3r \!"\\\ -5(aM\"5r"\#"\"5 \$"S\\ S\"35e M> \RJ\RL\RN\RP1r)\RL\RN1r*\RN\RV\RX\RZ\RL\RJ\RP\R\\R^\R`\Rb\Rd\Rf\Rh\Rj\Rl\Rn\Rp\Rr\Rt\Rv\Rx\Rz\R|\R~\R/rASgSjrBSgSjrCSgSjrDSgSjrESgS jrF\CS!\DS"0rG"S#S$\5rH0rIS%\JS&'\"\HR\HR\HR\HR\HR/5\IS''\"\HR\HR/5\IS('\"\HR\HR/5\IS)'\IS'\IS(-\IS*'\IS*\IS)-\IS+'\IS(\IS)-\IS,'"S-S.\5rT"S/S0\5rU"S1S2\5rV\"SS 9"S3S455rW"S5S6\5rX\"SS 9"S7S855rY\"SS 9"S9S:55rZ\"SS 9"S;S<55r[\"SS 9"S=S>55r\\"SS 9"S?S@55r]\"SS 9"SASB55r^\"SS 9"SCSD55r_"SESF\5r`\"SS 9"SGSH\_55ra\"SS 9"SISJ\_55rb\"SS 9"SKSL\_55rc\"SS 9"SMSN\_55rd\"SS 9"SOSP55re\"SS 9"SQSR55rf\"SS 9"SSST55rg\"SS 9"SUSV55rh\"SS 9"SWSX55ri/SYQrj\"SS 9"SZS[55rk\"SS 9"S\S]55rlShS^jrm\"SS 9"S_S`55rnSiSajroSjSbjrpSkScjrq\"SS 9"SdSe55rrgs snf)l) annotationsN) dataclass)autoEnum)CallableOptional TYPE_CHECKING) assert_neverNamespaceHelper OrderedSet)IteratorSequenceT)frozenc4\rSrSr%S\S'S\S'S SjrSrg) Location+strfileintlinec8URSUR3$)N:)rrselfs F/var/www/auris/envauris/lib/python3.13/site-packages/torchgen/model.py__str__Location.__str__0s))Adii[))Nreturnr)__name__ __module__ __qualname____firstlineno____annotations__r__static_attributes__rrrrr+s I I*rrc0\rSrSr\"5r\"5rSrg)Variant5rN)r"r#r$r%rfunctionmethodr'rrrr)r)5svH VFrr)z at::nativezVCPU CUDA HIP XLA MTIA MPS IPU XPU HPU VE Lazy Meta PrivateUse1 PrivateUse2 PrivateUse3) QuantizedSparse SparseCsr NestedTensorAutogradAutogradNestedTensorr2 quantizedquantized_decomposedc\rSrSrSr\r\"5r\"5r\"5r \"5r \"5r \"5r \"5r \"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r \"5r!\"5r"\"5r#\"5r$\"5r%\"5r&\"5r'\"5r(\"5r)\"5r*\"5r+\"5r,\"5r-\"5r.\"5r/\"5r0\"5r1\"5r2\"5r3\"5r4\"5r5\"5r6\"5r7\"5r8\"5r9\"5r:\"5r;\"5r<\"5r=\"5r>\"5r?\"5r@\"5rA\"5rB\"5rC\"5rD\"5rE\"5rF\"5rG\"5rH\"5rI\"5rJ\"5rK\"5rL\"5rM\"5rN\"5rO\"5rP\"5rQ\"5rR\"5rS\"5rT\"5rU\"5rV\"5rW\"5rX\"5rY\"5rZ\"5r[\"5r\\"5r]\"5r^\"5r_\"5r`\"5ra\"5rb\"5rc\"5rd\"5re\"5rf\"5rg\"5rh\"5ri\"5rj\"5rk\"5rl\"5rm\"5rn\"5ro\"5rp\"5rq\"5rr\"5rs\"5rt\"5ru\"5rv\"5rw\"5rx\"5ry\"5rz\"5r{\"5r|\"5r}\"5r~\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5rSSjrSSjr\S Sj5rSrg) DispatchKeySrcUR$Nnamers rrDispatchKey.__str__ yyrc4[U5R5$r:)rlowerrs rr@DispatchKey.lowers4y  rc~[RR5HupX:XdM Us $ [SU35e)Nzunknown dispatch key )r7 __members__itemsAssertionErrorvaluekvs rparseDispatchKey.parses<++113DAz44UG<==rrNr )rGrr!r7)r"r#r$r% UndefinedCatchAllrFPGAMAIAVulkanMetalMKLDNNOpenGLOpenCLIDEEPCustomRNGKeyId MkldnnCPUr/r0r1DensePythonTLSSnapshot PreDispatchPythonDispatcherPythonFuncTorchDynamicLayerBackMode ZeroTensor ConjugateNegative BackendSelectNamed AutogradOtherAutogradFunctionalityr3TracerAutocast AutocastCPU AutocastCUDABatchedVmapModeFuncTorchGradWrapperFuncTorchBatchedBatchedNestedTensorFuncTorchVmapModeFuncTorchDynamicLayerFrontMode FunctionalizeTESTING_ONLY_GenericWrapperTESTING_ONLY_GenericModeADInplaceOrViewr2CompositeImplicitAutograd%CompositeImplicitAutogradNestedTensorCompositeExplicitAutograd&CompositeExplicitAutogradNonFunctionalFuncTorchBatchedDecompositionCPUCUDAHIPXLAMTIAMPSIPUXPUHPUVELazyMeta PrivateUse1 PrivateUse2 PrivateUse3 QuantizedCPU QuantizedCUDA QuantizedHIP QuantizedXLA QuantizedMTIA QuantizedMPS QuantizedIPU QuantizedXPU QuantizedHPU QuantizedVE QuantizedLazy QuantizedMetaQuantizedPrivateUse1QuantizedPrivateUse2QuantizedPrivateUse3 SparseCPU SparseCUDA SparseHIP SparseXLA SparseMTIA SparseMPS SparseIPU SparseXPU SparseHPUSparseVE SparseLazy SparseMetaSparsePrivateUse1SparsePrivateUse2SparsePrivateUse3 SparseCsrCPU SparseCsrCUDA SparseCsrHIP SparseCsrXLA SparseCsrMTIA SparseCsrMPS SparseCsrIPU SparseCsrXPU SparseCsrHPU SparseCsrVE SparseCsrLazy SparseCsrMetaSparseCsrPrivateUse1SparseCsrPrivateUse2SparseCsrPrivateUse3NestedTensorCPUNestedTensorCUDANestedTensorHIPNestedTensorXLANestedTensorMTIANestedTensorMPSNestedTensorIPUNestedTensorXPUNestedTensorHPUNestedTensorVENestedTensorLazyNestedTensorMetaNestedTensorPrivateUse1NestedTensorPrivateUse2NestedTensorPrivateUse3 AutogradCPU AutogradCUDA AutogradHIP AutogradXLA AutogradMTIA AutogradMPS AutogradIPU AutogradXPU AutogradHPU AutogradVE AutogradLazy AutogradMetaAutogradPrivateUse1AutogradPrivateUse2AutogradPrivateUse3rr@ staticmethodrJr'rrrr7r7SsIH 6D 6D VF FE VF VF VF FEVNI VFI6L FE&Kv VF$(F!JIvHFM FEFM F6 VFvH&K6LfGvH6v&%)V"FM"&&#vfOvH $,0F) $-1V*$(F! &C 6D &C &C 6D &C &C &C &C B 6D 6D&K&K&K6LFM6L6LFM6L6L6L6L&KFMFM666IJIIJIIIIvHJJ6LFM6L6LFM6L6L6L6L&KFMFM666fOvfOfOvfOfOfOfOVNvv"f"f"f&K6L&K&K6L&K&K&K&KJ6L6L&&&!>>rr7c>\rSrSr\"5r\"5r\"5rSrg)_TorchDispatchModeKeyrN) r"r#r$r%rFAKEPROXY FUNCTIONALr'rrrrrs 6D FEJrrc^/n[H"mURU4Sj[55 M$ SRU5$)Nc34># UH nSTUS3v M g7f)z z = auto()Nr).0bcfks r .codegen_per_backend_entries..s G4Fb4tB4y)4F )FUNCTIONALITY_KEYSextendBACKEND_COMPONENTSjoin)rrs @rcodegen_per_backend_entriesrs3A  G4FGG! 99Q<rzMissing zL from DispatchKey enum. Here is the autogenerated list we expect to have: cU[R[R[R[R1;$r:)r7rvrwrtrudks ris_generic_dispatch_keyr7s8 --::--99  rcU[R[R[R[R[R [R 1;$r:)r7rzrrrrrrs ris_cuda_dispatch_keyrAsJ !!!!$$   rcU[R[R[R[R[R [R 1;$r:)r7rrrrrrrs ris_xpu_dispatch_keyrMsH     ##  rcU[;$r:)STRUCTURED_DISPATCH_KEYSrs ris_structured_dispatch_keyrZs ) ))rcU[;$r:)UFUNC_DISPATCH_KEYSrs ris_ufunc_dispatch_keyr^s $ $$rcudaxpucV\rSrSr\"5r\"5r\"5r\"5r\"5r \"5r \"5r \"5r \"5r \"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5rSSjr\S Sj5r\S Sj5r\S Sj5rSrg) ScalarTypeigcUR$r:r;rs rrScalarType.__str__{r>rcd[RR5HupX:XdM Us $ gr:)rrCrDrFs r maybe_parseScalarType.maybe_parse~s,**002DAz3rcJ[RU5nUc SU35eU$)Nzunknown dtype )rr)rGmb_rs rrJScalarType.parses/%%e,9>%!99 rc[5nURS5HKnU[;aUR[U5 M'UR [ R U55 MM U$)N, )r split DTYPE_CLASSESupdateaddrrJ)valuesdtypesrGs r parse_setScalarType.parse_setsS)3\\$'E % mE23 :++E23 (  rrNr )rGrr!zScalarType | None)rGrr!r)r rr!OrderedSet[ScalarType])r"r#r$r%rByteCharShortIntLongHalfFloatDouble ComplexHalf ComplexFloat ComplexDoubleBoolBFloat16 Float8_e5m2Float8_e5m2fnuz Float8_e4m3fnFloat8_e4m3fnuzFloat8_e8m0fnurrrrJr r'rrrrrgs 6D 6D FE &C 6D 6D FE VF&K6LFM 6DvH&KfOFMfOVN  rrz!dict[str, OrderedSet[ScalarType]]rIntegralFloatingComplexAll AllAndComplexFloatingAndComplexc\rSrSr\"5r\"5r\"5r\"5r\"5r \"5r \"5r SSjr \ SSj5rSrg)UfuncKeyicUR$r:r;rs rrUfuncKey.__str__r>rc~[RR5HupX:XdM Us $ [SU35e)Nzunknown ufunc key )r'rCrDrErFs rrJUfuncKey.parses<((..0DAz11%9::rrNr )rGrr!r')r"r#r$r%r CUDAFunctorCUDAFunctorOnOtherCUDAFunctorOnSelf CPUScalar CPUVector ScalarOnlyGenericrrrJr'rrrr'r'sR&KIIJfG;;rr'c\rSrSrSrSrSrg)DeviceCheckTypeirrN)r"r#r$r%NoCheck ExactSamer'rrrr4r4s GIrr4c>\rSrSr\"5r\"5r\"5rSrg)ViewSchemaKindirN) r"r#r$r%raliasingaliasing_inplace non_aliasingr'rrrr9r9svHv6Lrr9c\rSrSr%S\S'S\S'S\S'S\S'S \S 'S \S 'S \S 'S\S'S\S'S\S'S\S'S\S'S\S'S\S'S\S'S \S'S\S'S\S'S\S 'S\S!'S\S"'S\S#'S\S$'S\S%'\S0S1S'jj5rS2S(jrS2S)jr\ S3S*j5r \ S3S+j5r \ S4S,j5r \ S5S-j5r \ S3S.j5rS/rg&)6NativeFunctionir namespaceFunctionSchemafuncbool!use_const_ref_for_mutable_tensors device_guardr4 device_check str | None python_modulecategory_overridez set[Variant]variantsmanual_kernel_registrationmanual_cpp_bindingrloczlist[OperatorName]autogenzdict[UfuncKey, UfuncInnerLoop]ufunc_inner_loop structuredzOperatorName | Nonestructured_delegatestructured_inheritszPrecompute | None precomputedset[str]cpp_no_default_args is_abstract&has_composite_implicit_autograd_kernel4has_composite_implicit_autograd_nested_tensor_kernel&has_composite_explicit_autograd_kernel5has_composite_explicit_autograd_non_functional_kerneltagsNcHUR5nURS5n[U[5(d SU35e[R "USS9nUR SS9n[RUR5nURS/5n [U [5(de[U 5n URSS 5n [U [5(deU (a"URR(aS 5eURS S 5n [U [5(de[5n U RS 5H]nUS :Xa!U R!["R$5 M*US:Xa!U R!["R&5 MQ[)SU35e URSS 5n[U[5(d SU35eURSS 5n[U[5(d SU35eURSS5n[U[5(d SU35eURSS5nUb[U[5(d SU35eUbU[*R,;d SU35eUc[*R.nO [*UnURSS 5n[U[5(d SU35eURSS5nUb[U[5(d SU35eUb SU;dS5eSnUb[0RU5nURSS5nUb[U[5(d SU35eUb SU;dS5eURSS5nUb[U[5(d SU35eUb["R&U ;dS5eURS S5nUb[U[5(d SU35eURS!S5nUbUSLdeU(a[2RU5OSnURS"/5n[U[5(aU/n[U[5(deUS:XaS#U;aUR5S#5 [5nUH9n[7U5S$:deUU;aUR!U5 M-[)S%U35e S$S&KJn URS'S5n U b[U [<5(dU5e0n!S$n"U GbU(aS(5eS n#U R?5GH9un$nU$S):XaM[U$[5(d S*U$S+U 35e[U[5(d S,US+U 35eU$RS-5Hn%[@RU%RC55n&U"S- n"U(aU&U;aM:U&[D;dS.U&S/US035e[R "US1S29nUR S3S9n'[GURU=(a [IU&5U'S4-S59U!U&'U&[@RJLdMUURMU5:XdMSn#M GM< U"S:XaU#(aS65eU(d}U!RO5[@RJ1:wdZU![@RJRQ5(d4U"S:wd.S7URMU5S8U![@RJS935eOU(dUc[URLRL5n(U(RSS:5(dPU(RUS;5(d:URRV(a*URRY5(d S9U![@RJ'U!V)s/sHWn)U)[@R\:Xd>U)[@R^:Xd*U)[@RJ:XdU)[@R`:XdMUU)PMY n*n)[7U*5S::d>[7U*5S?:Xa([@R^U*;a[@R`U*;dS@5eURSASB5n+[U+[5(deU+SB:Xa/O4U+RS 5V,s/sHn,[0RU,5PM sn,n-URSC05n.0n/[U.[5(a6[bRU.[dRf5U/[dRf'O[U.[<5(aU.R?5Hzun%n0U%S):XaM[U%[5(d SDU%35e[U0[5(d SEW35e[dRU%5n1[bRU0U15U/U1'M| O[)SFU.35eU/(aVU(dSG5eS$SK4J5s J6n2 [nH5n&U&U!;d SHU&35e[GU2RqUU&5S[ZS59U!U&'M7 U(aSn3OU!RO5[@RJ1:g=(aZ U!RO5[@R`1:g=(a1 U!RO5[@RJ[@R`1:gn3[@RJU!;n4[@R`U!;n5[@R\U!;n6[@R^U!;n7U!R?5V%Vs0sHun%nU%URLU0_M n8n%nURS)S5 U(a SIU35eUb[rHn9U9U!;aM SJU9SK35e [uSS0SU_SU _S U _SU_SU_SU_S!U_SAU-_SCU/_SU_SU_SU_S U_SU_SU_SLU_SU _SMU3_SNU4_SOU5_SPU6_SQU7_S"U_SRU_6U84$s sn)fs sn,fs snn%f)TzX Parse a NativeFunction from a dictionary as directly parsed from native_functions.yaml rAz not a str: r5)namespaced_entity max_levelaten)defaultrTrCFz4see https://github.com/pytorch/pytorch/issues/145522rIr+rr,zillegal variant rJz not a bool: rKrDTrENzillegal device_check: rOrP::zbnamespace is not supported in structured delegate, using the same namespace as the native functionrQzbnamespace is not supported in structured inherits, using the same namespace as the native functionrGz&functions in modules cannot be methodsrHrRrZpt2_compliant_tagrz illegal tag )cppdispatchzncannot specify both manual_kernel_registration and dispatch; with manual registration, dispatch has no effect!__line__zillegal dispatch key 'z' in zillegal dispatch value ',z Dispatch key z of kernel z! is not a supported dispatch key.)r]atz::native)kernelrO cpp_namespacezSunnecessary dispatch table for this function; just delete the dispatch key entirelyzQunexpected name for singleton CompositeImplicitAutograd dispatch entry: expected z but got zR. Rename your implementation to the expected name, then delete the dispatch tablenew__likez expected z to have a CompositeExplicitAutograd dispatch entry, but there was no dispatch table. Factory functions should not have implicit dispatch as they should not be decomposed for __torch_dispatch__)rOriaHcannot specify more than one of CompositeExplicitAutograd, CompositeExplicitAutogradNonFunctional, or CompositeImplicitAutograd on a single kernel; each strictly subsumes the other. If you wanted to provide an explicit autograd implementation, specify CompositeExplicitAutograd; otherwise specify CompositeImplicitAutograd onlyrMr-rNz#ufunc_inner_loop key is not a str: z%ufunc_inner_loop value is not a str: z"ufunc_inner_loop not str or dict: zufunc must be structuredz2ufunc should not have explicit dispatch entry for zleftover entries: z+if structured_delegate, then must not have z* in dispatch dictionary (it is delegated!)rLrUrVrWrXrYr?r);copypop isinstancerr from_namespaced_entityget_cpp_namespacer@rJ entity_namelistsetrB argumentsoutrrr)r+r,rEr4rCr7 OperatorName Precomputeappendlen torchgen.apirbdictrDr7strip dispatch_keysBackendMetadatarrtr<keyssupports_symint startswithendswithtensor_optionshas_tensor_argDEFAULT_KERNEL_NAMESPACErvrwruUfuncInnerLoopr'r2torchgen.api.ufuncapiufuncrschema_kernel_namerr>):eirL valid_tags ignore_keysefuncsnamespace_helperr?rAcpp_no_default_args_listrTrC variants_srIrIrJrKrDdevice_check_srErOstructured_delegate_srPrQrGrHprecomputed_dictrRtags_inprZtrb raw_dispatchrcnum_dispatch_keys%redundant_composite_implicit_autogradksrH dispatch_keykernel_namespacer<dcomposites_in_dispatch autogen_strxrMraw_ufunc_inner_looprNvo ufunc_keyrrUrVrWrXrYbackend_metadatakeys: r from_yamlNativeFunction.from_yamlXsH  GGIf %%%<UG'<<%*AA#q %66v6F ##$4$@$@A#$55)>#C 2D9999!":;,-EE /- );TBBBB ,~~)) F )UU:z2 *c****!$!!$'AJ W--.h W^^,$'7s%;<< (&'UU+G%O"4d;; 56 7 ;UU#7?,d33X|DVCW5XX3uu^T2 ,--Ll^/LL-~t4%NC)H)H .) * H  "n8S8S&S 5 #N#3 4 5 S  !*44L*>:LUU</ *d++H|J<-HH+ !&;T B$, !31 1  1 ./ 0 1 %, >( J >;C..:JK:J$!QA 1~-:JK j$.*1#..u  */(*A#G))*0   2S " &   %8   %8  (   "2 ,F $6 , #4 * *  ! "%8# $(% &8^' (Fz) *8^+ ,G|- ./ 0$1 4 7  k" >JNLs1Ap p"ppcdUR(aS5eUR(dS5eg)NzMThis function is structured, but there was no valid functional variant of it.zThis function delegates to another structured out function, but no valid function was found (the delegate may not exist, or it has the wrong type)rOrPrs rvalidate_unstructured$NativeFunction.validate_unstructureds;??  1 "''  e 'rcURRR(a&UR[R 1:XdS5eUR (aKURR5[R:XdS5eUR(dS5eUR(aKURR5[R:wdS5eUR(dS5eUR (aUR(aS5eURR5Vs1sHoRcMURiM nn[RUR U5n[#U5S:Xd SU35eUR$bUR (dS5e['URR5R)S 5(a%UR*[,R.:XdS 5eS ['URR5;dS ['URR5;d5[1S URRR255(aPS['URR5;a-['URRR5S;d)URRR55(a5SUR6;d$['URR55eggs snf)NzNative functions with out arguments MUST be declared with only function variant; e.g., variants: function; otherwise you will tickle a Python argument binding bug (which usually manifests itself as the result variable being undefined.)zYPut structured field on the out= variant of a function; did you mean structured_delegate?z:device_guard: False is not respected by structured kernelszQstructured_delegate field not allowed on out= functions; did you mean structured?z?Cannot have both structured and structured_delegate on functionrzInvalid cpp_no_default_args: z4structured_inherits must also imply structured: True_foreachzsforeach kernels fall back to slow path when tensor are on different devices, device_check not allowed to be enabledranddropoutc3@# UHnSUR;v M g7f)rNr;rargs rr/NativeFunction.__post_init__..s9U# SXX-9Usbackward)_cudnn_init_dropout_statenondeterministic_seeded)rArurvrIr)r+rOkind SchemaKindrDrPschema_order_argumentsr_r<rt differencerTrzrQrrrEr4r6anyflat_allhas_generator_argrZ)radefaulted_arguments invalid_argss r __post_init__NativeFunction.__post_init__s 99   " "==W%5%5$66 [ 6 ??99>>#z~~5 K 5$$ L $  # #99>>#z~~5 > 5$$ L $ OO(@(@ M A!II<<> >q))FAFF>  ~~d&>&>@ST < A%U)F|n'UU%  # # /?? F ? tyy~~  ) )* 5 5$$(?(?? 9 ? c$))..) )TYY^^!449=9L9L9U9U c$))..&99 ++,4QQyy""4466, 9 N3tyy~~;N N97= s MMcUR=(d UR=(d UR=(d UR=(a UR$r:)rVrXrYrWrs rhas_composite_kernel#NativeFunction.has_composite_kernelsO  7 7 J:: JII  7 7 JII  rcURRn[U5S:=(a [SU55nSUR;=(aK [ URR 5S:g=(a" [ URR 5S:gn[SURR555nU=(d U=(d U$)Nrc3# UH4oRSL=(a URR(+v M6 g7fr: annotationis_writerrs rr,NativeFunction.is_view_op..*s/5 LPqLL $ BQ\\-B-B)B BDs<> inplace_viewresize_ resize_as_c3~# UH3nURSL=(a SURR;v M5 g7f)N*)ralias_set_after)rinps rrr3s7 9 NN$ & P3#..2P2P+P P9s;=)rAreturnsrzrrZrr<r)rretsis_non_mutating_viewis_inplace_viewis_wildcard_views r is_view_opNativeFunction.is_view_op'syy  "4y1} 5 LP5 2  dii ' 4DIINN#y0 4DIINN#|3   yy779  $JJ:JJrc(UR(aQURRRR(a"SUR;de[ R $UR(a[ R$[ R$)Nr) rrAr<inplacerZr9r;r:r<rs rview_schema_kindNativeFunction.view_schema_kind9s\ ??tyy~~22::!TYY. ..!22 2 ??!** *!.. .rcVURRRR$r:)rAr<basers r root_nameNativeFunction.root_nameCsyy~~""'''rcDUR=(d URSL$r:rrs rpart_of_structured_group'NativeFunction.part_of_structured_groupGsF$":":$"FFrrr:) rzdict[str, object]rLrrrSrzset[DispatchKey] | Noner!zMtuple[NativeFunction, dict[DispatchKey, dict[OperatorName, BackendMetadata]]]r!Noner!rB)r!r9r )r"r#r$r%r&rrrrpropertyrrrrrr'rrrr>r>s} N (,+"!"! !%$  M 54-, $##""!-10:>>,00;?? N 04 o o o o - o W o o b  &=O~  KK"//((GGrr>cZ\rSrSr\"5r\"5r\"5r\"5r\"5r Sr g)riLrN) r"r#r$r%r functionalrrvmutablescratchr'rrrrrLs%JfG &CfGfGrrc\rSrSr%S\S'S\S'S\S'S\S'\SSj5rSS jrSS jrSS jr \SS j5r \ SS j5r Sr g)NativeFunctionsGroupiZr>rNativeFunction | Nonerrrvc.URR$r:)rvrOrs rrONativeFunctionsGroup.structuredasxx"""rc  URRR5nUR5HnXRR5:wa)[ SUSURR535eUR UR :wdMe[ SURRRSURR35e URRR5[R:XdeURRR5[R:XdeURRURR:XdeURbhURRR5[R:XdeURRURR:XdeURbURRR5[R:XdeURRURR:XdeURRRRR(deUR (GaURR(dURR (aeURR"URRR:Xd_URRRSURR"SURRR35eURb:URR"URRR:Xde[%UR5Vs/sH4nSUR&;dM[)URR5PM6 sn5nSR+S U55n[-5nUR5H%nUR/S UR055 M' SR+S [%U555n[3U5S :Xa@[3U5S :a1[5S USUS[)WRR535eXd:wa[5S USUSUS35egs snf)Nz_NativeFunctionsGroup constructed from two NativeFunctions that don't have matching signatures:  != zMNativeFunctionsGroup constructed from structured and unstructured functions:  and z delegates to z but its actual delegate is generatedrc38# UHn[U5v M g7fr:rrrs rr5NativeFunctionsGroup.__post_init__..s%Dmc!ffmc38# UHn[U5v M g7fr:r)rops rrrs)FIb#b''Irc38# UHn[U5v M g7fr:rrs rrrs/ :qCFF:rrz,The codegen expects to be able to generate 'zv'. In order to generate them however, we expect them to be called out explicitly in the yaml. Please add an 'autogen: z' line to the entry for z*'. To do so, it expects a line: 'autogen: z'. Instead, it found 'autogen: ')rrA signature functionsrErOrrvr<rrr?rrfunctional_overloadrVrWrPsortedrZrrrtrrMrz RuntimeError)rtest_sigf generated_fnsgenerated_fns_strexpected_generated_fnsexpected_generated_fns_strs rr"NativeFunctionsGroup.__post_init__fs #'??#7#7#A#A#C!A66++--$<,>>>> << #<<$$))+z/A/AA AA<<))T__-F-FF FF << #<<$$))+z/A/AA AA<<))T__-F-FF FF??'',,11EE EE ???HHCCUU V??66$((--:L:LL ??'',,-^DOO<_<_;`a..2hhmm.@.@-AC L||'||77488==;M;MMMM'+~~'7 Q'7!;!&&;P S '7 Q !II%Dm%DD+.5!A " ) ))FAII)F F"%)YY/ "#9:/ & " % &! +M0BQ0F>?P>QR,,=+>>VWZ[\[a[a[f[fWgVhj  & :>?P>QR;;L:MN00J/K1N  ; Rs #T ;#T cJURRR5$r:)rvrArrs rrNativeFunctionsGroup.signaturesxx}}&&((rc## URv URv URbURv URbURv gg7fr:)rrvrrrs rrNativeFunctionsGroup.functionssEoohh << #,,  << #,,  $sAAc.URR$r:)rrrs rrNativeFunctionsGroup.root_names(((rcU(de[U5S:Xag[U5nUR[RS5nUR[R S5nUR[R S5nUR[RS5nU(aeUceUcg[UUUUS9$)Nr5)rrrrv) rzr|rnrrrrrvr)rrrrrvs r from_dictNativeFunctionsGroup.from_dictsq q6Q; GUU:00$7 %% **D1%% **D1eeJNND) u%%% ;#!   rrNrrr!r@)r!Iterator[NativeFunction]r )rz dict[SchemaKind, NativeFunction]r!zNativeFunctionsGroup | None)r"r#r$r%r&rrOrrrrrrr'rrrrrZsd "" ""  ##?B)))  rrc>\rSrSr%S\S'S\S'S\S'S SjrSrg ) rirrhrBrOric SUR;$)N_symint)rhrs rrBackendMetadata.supports_symintsDKK''rrNr)r"r#r$r%r&rr'rrrrrs K (rrcH\rSrSr%S\S'S\S'S\S'\S Sj5rS rg ) rirr<r supported_dtypesr'rcURSS5up#USS:XdeUSS:Xde[5nUSSRS5HnU[RU5-nM [ X$US9$) N r5r()r)r<rr)rr rr r)rGrr<supported_dtypes_strrrHs rrJUfuncInnerLoop.parses%*[[a%8"#A&#---#B'3...3=<%a+11$7A  4 4Q 7 7 8I  rrN)rGrrr'r!r)r"r#r$r%r&rrJr'rrrrrs' I,,    rrc\rSrSr%S\S'S\S'S\S'S\S'S\S '\SS j5rSS jrSS jrSS jr SSjr Sr g) BackendIndexir7rrBuse_out_as_primaryrDexternalz#dict[OperatorName, BackendMetadata]indexcUR5H:up#UR5H!upEX@U;d SUSU35eXPUU'M# M< g)Nzduplicate operator z for dispatch key )rD) parent_index child_indexrHrIop_namemetadatas r grow_indexBackendIndex.grow_indexsb  %%'DA%&WWY!1o5)'2DQCH5,4Q( &/(rcTUR(a UR$UR$r:)r(rvr)rgs rprimaryBackendIndex.primarys  " "55L<< rc,URU5nUSL$r:) get_kernel)rr3ms r has_kernelBackendIndex.has_kernel#s OOA }rc&[U[5(aUnO2[U[5(aURU5nO [ U5 WR R UR;agURUR R $r:)ror>rr4r rAr<r*)rr3rs rr7BackendIndex.get_kernel'sf a ( (A / 0 0 QA O 66;;djj (zz!&&++&&rcVUR(a[UR5S3$g)NNativeFunctions)r)rrrs rnative_function_class_name'BackendIndex.native_function_class_name4s' ==$++,-_= = rrN)r,6dict[DispatchKey, dict[OperatorName, BackendMetadata]]r-rAr!r)r3rr!r>)r3%NativeFunction | NativeFunctionsGroupr!rB)r3rBr!zBackendMetadata | None)r!rF) r"r#r$r%r&rr0r4r9r7r?r'rrrr'r'ssN .. 4L 4K 4  4 4  '6 '  'rr'c&\rSrSr%S\S'S\S'S\S'\SSj5rSS jr\R"S 5r \ SS j5r SS jr SS jrSSjrSSjrS SjrS!SjrSSSS.S"SjjrS#SjrS$Sjr\SSj5rSSjrS%SjrSrg)&r@isrwr< Argumentsrutuple[Return, ...]rc`^SSjm[U4SjURR55$)NcJURcgURR$NFr)rs rr+FunctionSchema.is_mutable..is_writes~~%>>** *rc34># UH nT"U5v M g7fr:r)rrrs rr,FunctionSchema.is_mutable..s@(?18A;;(?r)rArgumentr!rB)rrur)rrs @r is_mutableFunctionSchema.is_mutable}s% +@(?(?@@@rc[R"URRURRURR 5$r:) itertoolschainruflat_positionalflat_kwarg_onlyrvrs rr%FunctionSchema.schema_order_argumentss: NN * * NN * * NN    rz3(?P[^\(]+)\((?P.*)\) -> (?P.*)cL[RRU5n[U5S:Xd SU35eUSup#n[R U5n[ R U5n[U5n[XVUS9n[U5U:Xd[U5SU35eU$)Nr5zInvalid function schema: rr<rurr) r@decl_refindallrzrwrJrD parse_returnsr) rAdeclsopsargs return_declr<rurrs rrJFunctionSchema.parses&&..t45zQB";D6 BB!&q;!!#&OOD)  , 7 K1v~4#a&dV44~rc:[SUR55$)Nc3~# UH3nURcMURR(dM/Uv M5 g7fr:rrs rr5FunctionSchema.returns_are_aliased..s2 !|| ,-LL,A,A A!s == =)rrrs rreturns_are_aliased"FunctionSchema.returns_are_aliaseds  \\   rc ^^[URRUR5H'unmURTR:XaM"S5e URR H:m[ U4SjUR55(dM)S[U535e [URR5URRVs/sHoRS:XdMUPM sn-nURVs/sH1nURcMURR(dM/UPM3 nnURVs/sH/nURbURR(aM-UPM1 nn[U5S:Xd"[U5S:XdS[U535eUH$m[ U4SjU55(aMS5e URR(a[ S URR55(a![UR5S:XdS 5eOp[URRVs/sH'nURRS 5(aM%UPM) sn5[UR5:XdS 5eURRR(aURRnU(a@UR R(a%UR RR(deUR R"[%[&R(5:XaM[UR5S :Xa1URSRUR R:XdeO[UR5S:XdeURR*b5UR-5[.R0:XdS[U535eUR35(a6UR-5[.R0:XdS[U535eggs snfs snfs snfs snf)NzsOut arguments must have matching return Tensor; furthermore, the ith-argument needs to correspond to the ith returnc3V># UHnTRUR:Hv M g7fr:rrrrs rr/FunctionSchema.__post_init__..sNA1<<1<<7&)z^If you have a schema with mutable positional args, we expect them to not be returned. schema: rrzYNativeFunctions must have either only mutable returns, or only immutable returns. Found: c3V># UHnTRUR:Hv M g7fr:rf)rrrets rrrhsPP66Xfmm44>Ps24z3out= ops that accept tensor lists as out arguments _scratch_zMMust return as many arguments as there are out arguments, or no return at allr5zFound an operator that is not functional or out variant, but has tensor options arguments.This is not allowed- tensor options arguments are only allowed for factory functions.schema: zFound an operator that is not functional, but its overload contains the string 'functional'.This is a special keyword in the codegen, please use a different overload name.schema: )ziprurvrrpost_self_positional_mutablerrrsrRr<rrzrrself_argargumentrmrnrorprrrris_functional_fn)rr out_and_selfrkmutable_returnsimmutable_returnsself_ars ` @rrFunctionSchema.__post_init__sDNN.. =HC>>S^^3 I 3><>993 9CXX=OC93  || #~~ .1nn.E.E #  || #~~%S^^-D-D #  ?#q(C0A,Ba,G ghklphqgr s G#CP<PPP J P# >>   Qdnn>P>PQQQ4<<(A-I-c$(>>#5#5#5C"xx22;?#5 & 'd' 99>> ! !^^,,FOO..OO..77 8##x '>> %* Q22foo6P6PPQP 4<<(A--- >> ( ( 499;*"7"77 t9+' 7  " "99;*"7"77 t9+' 7 #W3   Bs6"Q99Q9Q>'Q>Q>,R R1$RRc4SURR;$)Nrr< overload_namers rrwFunctionSchema.is_functional_fnstyy6666rc@[URR5$r:)rBrurvrs r is_out_fnFunctionSchema.is_out_fn s6DNN&&''rc[URR5n[URRVs/sH&o"RR S5(dM$UPM( sn5nURRR n[ SURR55nU(a U(aeU(a[R $U(aU(dS5e[R$U(aU(aS5e[R$U(a[R$[R$s snf)z What kind of schema is this? A functional schema is one that returns a newly allocated output; an inplace schema modifies the self argument inplace; an out schema writes the result into an explicitly provided out argument. rrc3x# UH0nURSL=(a URRv M2 g7fr:rrqs rr&FunctionSchema.kind..4s2 8 LL $ >)>)> >8s8:zFinvariant: all scratch operators are expected to be out= operators toozjWe should not categorize a scratch op as an out variant. Check if the order of if statements are expected!) rBrurvr<rrrpost_self_positionalrrrr)ris_outr is_scratch is_inplacerMs rrFunctionSchema.kind(sdnn(() NN.. S.S((2E2Ek2RS. S YY^^++  ^^88  z** %% %  X 6%% % ! | >>> ! %% %(( (= Ts #E$Ec/nURHnURRVs/sH0nURcMURUR:XdM.UPM2 nn[ U5S:XaUR S5 Mv[ U5S:Xa UR USR 5 MSRSU55n[SUR SUS35e U$s snf)Nrr5rc38# UHoRv M g7fr:r;rqs rr6FunctionSchema.aliased_return_names..bs)G,Q&&,rzFound a return (z)that aliases multiple inputs (r#) rrurrrzryr<rrE)routsrr aliased_args aliased_namess raliased_return_names#FunctionSchema.aliased_return_namesUs!#A000A<<01  0L0  < A% D!\"a' LO001 $ )G,)G G $&qvvh.Mm_\]^ sC,C,C,F) strip_defaultstrip_view_copy_namekeep_return_namesc ^^SU4SjjnTRRRnU(aQURS5(aURSS5nO(URS5(aURSS5n[ UU4Sj[ R "TRRb!TRRR/O/TRRTRR555n[ [UTR55nXv-nTRRUS9n [TR5S :Xa.[ R#[U 5RS S 55n [%['[)US TRRR*S 9SS9U US9$)a Certain schemas are 'related', in that they are simply inplace/out/functional versions of the same function. This method factors these schemas into the "core" functional signature which is equal across all versions. Here is what normalization happens to the schema to convert it to a signature: - The overload name is stripped (name is retained, since it expresses semantic content about what the function does) - Inplace is set False - Out arguments are stripped - Mutable post_self_positional args are converted to returns - Mutability annotations are stripped (this is sound because you cannot overload on mutability annotation) - Return names are stripped since they are not overloadable and some variants have return names but some not - TensorOptions are dropped because out= variants of factory functions don't include them (and we want to be able to pair up factory functions with their out variants) Finally, we want to be able to pair up related "view" and their corresponding "view_copy" operators. We do this by optionally stripping the trailing "_copy" from the base name. Example of a mutable op before and after: f.func (Mutable operator): _fused_moving_avg_obs_fq_helper(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor(a!) running_min, Tensor(b!) running_max, Tensor(c!) scale, Tensor(d!) zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> (Tensor output, Tensor mask) # noqa: B950 f.func (Corresponding functional operator): _fused_moving_avg_obs_fq_helper.functional(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor running_min, Tensor running_max, Tensor scale, Tensor zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> (Tensor output, Tensor mask, Tensor running_min_out, Tensor running_max_out, Tensor scale_out, Tensor zero_point_out) # noqa: B950 f.func.signature() output: _fused_moving_avg_obs_fq_helper(Tensor self, Tensor observer_on, Tensor fake_quant_on, Tensor running_min, Tensor running_max, Tensor scale, Tensor zero_point, float averaging_const, int quant_min, int quant_max, int ch_axis, bool per_row_fake_quant=False, bool symmetric_quant=False) -> (Tensor, Tensor, Tensor, Tensor, Tensor, Tensor) # noqa: B950 cT>[T(a UROSURSS9$)Nr<rmr)Returnr<rm)rrs rstrip_ret_annotation6FunctionSchema.signature..strip_ret_annotations%0QVVdVV r_copyr-_scatterscatterinversec3 >^# UHmTRcMTRR(dM/[U4SjTR55(aMU[ T(aTR S3OSTR SS9v M g7f)Nc3V># UHnTRUR:Hv M g7fr:rfrgs rr5FunctionSchema.signature...sKl  4lri_outr)rrrrrr<rm)rrrrs @rr+FunctionSchema.signature..st,  ||! " %%# $KdllKK% F(9xt_tVV  sBB"B2Brz bernoulli.pzfloat pz float p=0.5Frr dunder_methodr~rV)rrr!r)r<rrreplacetuplerPrQrururvrvrmaprrrrDrJr@rwBaseOperatorNamer) rrrrr base_namereturns_from_mutable_inputsoriginal_returnsrargs_sigs ` ` rrFunctionSchema.signaturehs{X IINN'' !!'**%--gr: ##J//%--iC ',, __ ~~..:^^,,556""33 , ' #0!%94<>++-+H tyy>] * s8}'<'"> !   rc URSS9$)NTr)rrs rview_signatureFunctionSchema.view_signatures~~4~88rc@[UURURS9$)NrV)r@rur)rr<s r with_nameFunctionSchema.with_names nnLL  rcUR5[R[R[R4;$r:)rrrrvrrs rmodifies_arguments!FunctionSchema.modifies_argumentss*yy{z11:>>:CUCUVVVrc6URR5$r:)ruhas_symint_argrs r has_symintFunctionSchema.has_symints~~,,..rc[UR5n[UR5S:Xa[URS5nO/SSR [ [UR55-S-nUR SUSU3$)Nr5rr!rr#z) -> )rrurzrrrr<)rall_arguments_strrs rrFunctionSchema.__str__st/ t||  !$,,q/*GDIIc#t||&<==CG))A/0gY??rrNr)r!zIterator[Argument])rArr!r@r)r!r)r!zlist[str | None])rrBrrBrrBr!r@r)r<rwr!r@r )r"r#r$r%r&rrMrrecompilerWrrJrbrrwrrrrrrrrrr'rrrr@r@ss   AA jjOPG   `D7(:')Z,$%*"' g g # g  g  g R9 WW/@rr@cR\rSrSr%S\S'S\S'S\S'\S Sj5rS SjrS rg ) Annotationiztuple[str, ...] alias_setrBrrc[R"SU5nUc SU35eURS5URS5(aURS5OS-n[UR S55nURS5S:HnU(a[ U5S:a S US 35eURS 5(a)[URS 5R S55OS n[ U5S:a[ U5S:a S US 35e[ X4US9n[U5U:Xd USU35eU$)Nz2^([a-z])(\|[a-z])*(!?)( -> (\*|[a-z](\|[a-z])*))?$zunrecognized alias annotation r5rlr-|rf!z,alias set larger than 1 is not mutable, got z instead.rzSbefore alias set and after alias set cannot be larger than 1 at the same time, got )rrrr)rmatchgrouprrrzrr)annr8 before_aliasrr after_setrs rrJAnnotation.parses: HHJC P}D >seDD}wwqz1771::QWWQZ2F ,,,S12 771:$Y!!3 :3%y I 456GGAJJE!''!***3/0B  %)c)nq.@ abeaffo p A I 1v}-4uo-}rcSRUR5nUR(aUS3nSRUR5nU(aUSU3nU$)Nrr -> )rrrr)rrrs rrAnnotation.__str__sTHHT^^, ==$+QI((4#7#78 $+T/):;IrrN)rrr!rr ) r"r#r$r%r&rrJrr'rrrrrs-N$$  Drrc\rSrSr\S Sj5r\S Sj5rSSjrSSjrSSjr SSjr SSjr SS jr SS jr S rg )Typei.cf[RU5n[U5U:Xd USU35eU$)Nr)r_parser)rrs rrJ Type.parse0s3 KKN1v{)qcaSM){rcH[R"SU5nUb-[[R UR S555$[R"SU5nUbZUR S5b[ UR S55OSn[[R UR S55US9$[R"SU5nUb[UR S55$[[U5$![an[SU35UeSnAff=f)Nz^(.+)\?$r5z^(.+)\[([0-9]+)?\]$rl)elemsizez-^__torch__\.torch\.classes\.([a-zA-Z0-9_.]+)$zunrecognized type ) rr OptionalTyperrJrrListTypeCustomClassTypernroKeyErrorr)rr8rrs rr Type._parse6s HH[! $ = 1771: 67 7 HH+Q / =&'ggaj&<3qwwqz?$DAGGAJ!7dC C HHEq I ="1771:. . @F1I& & @!3A378a ? @s1D D! DD!c[er:NotImplementedErrorrs rr Type.__str__I!!rc[er:rrbase_tys ris_base_ty_likeType.is_base_ty_likeQrrc@UR[R5$r:)rrorprs ris_tensor_likeType.is_tensor_likeT##FMM22rc@UR[R5$r:)rro Generatorrs ris_generator_likeType.is_generator_likeWs##F$4$455rc@UR[R5$r:)rroSymIntrs ris_symint_likeType.is_symint_likeZrrc[er:rrs r is_nullableType.is_nullable]rrc[er:rrs r is_list_likeType.is_list_like`rrrN)rrr!rr rror!rBrr!zListType | None)r"r#r$r%rrJrrrrrrrrr'rrrrr.sM @@$""363""rrc,\rSrSr\"5r\"5r\"5r\"5r\"5r \"5r \"5r \"5r \"5r \"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5r\"5rSrg)roierN)r"r#r$r%rrrrprDimname DimVectorfloatrrBLayoutDevice DeviceIndexScalar MemoryFormatQSchemeStorageStreamrSymBool GraphModuler'rrrroroesIJ VF &CfGI FE &C 6D VF VF&K VF6LfGfG VF VFfG&KrrocR\rSrSr%S\S'S SjrS SjrS SjrSSjrS Sjr S r g )rni|ror<c0URR$r:r;rs rrBaseType.__str__s))..!"rc URU:H$r:r;rs rrBaseType.is_base_ty_likesyyG##rcgrHrrs rrBaseType.is_nullablercgr:rrs rrBaseType.is_list_likerc<UR[R:H$r:)r<rorrs rrBaseType.is_symint_likesyyFMM))rrNr rrr) r"r#r$r%r&rrrrrr'rrrrnrn|s L#$*rrncR\rSrSr%S\S'S SjrS SjrS SjrS SjrSSjr S r g )rirrc URS3$)N?)rrs rrOptionalType.__str__s))Arc8URRU5$r:rrrs rrOptionalType.is_base_ty_likeyy((11rc6URR5$r:rrrs rrOptionalType.is_symint_likeyy''))rcg)NTrrs rrOptionalType.is_nullablerrc6URR5$r:)rrrs rrOptionalType.is_list_likesyy%%''rrNr rrr r"r#r$r%r&rrrrrr'rrrrrs J2*(rrcR\rSrSr%S\S'S SjrS SjrS SjrS SjrSSjr S r g )rir class_namec SUR3$)z; Return the class name will prefix __torch__.torch.classes z__torch__.torch.classes.)r-rs rrCustomClassType.__str__s*$//):;;rcgrHrrs rrCustomClassType.is_base_ty_likerrcgrHrrs rrCustomClassType.is_symint_likerrcg)z( Assume a custom class is not nullable. Frrs rrCustomClassType.is_nullablesrcgr:rrs rrCustomClassType.is_list_likerrrNr rrrr+rrrrrs O<  rrc\\rSrSr%S\S'S\S'S SjrSSjrSSjrSS jrSS jr S r g )rirrz int | NonercfUR(a UROSnURSUS3$)Nr-[])rr)rrs rrListType.__str__s,!%$))))AdV1%%rc8URRU5$r:r rs rrListType.is_base_ty_liker"rc6URR5$r:r$rs rrListType.is_symint_liker&rc6URR5$r:)rrrs rrListType.is_nullablesyy$$&&rcU$r:rrs rrListType.is_list_likes rrNr rrrr+rrrrrs& J &2*'rrc\rSrSr%S\S'S\S'S\S'S\S '\SS j5r\SS j5r\SS j5r SS jr Sr g)rLirr<rrmrFr_Annotation | NonercUR$r:rfrs r alias_infoArgument.alias_info rczSU;d SUS35eSU;aOURS5S:Xd SUS35eURS5upURSS5up4USU3nOURSS5up5UnSn[R"SU5nU(aUUR S5S ;dS 5eS UR S5-n[ RUR S55nOUnSn[RU5n [UU UUS 9n [U 5U:Xd[U 5S U35eU $)Nr zillegal argument 'r=r5z&illegal argument with default value: 'Tensor\((.+)\)(.*)rlr-rz[],unrecognized alias analysis form with Tensorrpr<rmr_rr) countrrsplitrrrrrJrrLr) rtype_and_annot_and_namer_type_and_annotr<name_and_defaultrtype_srrmrs rrJArgument.parsesdcz6/uA66z #:99S>Q& W*PQTPUUV(W W&/2yy~ , ##:#A#A#q#I N"&q 2 /2zz#q/A ,N#DG.? ;;q>& >>  >  A.F#))%++a.9J#FJzz&! !   1v}2QSE22}rcXURSL=(a URR$r:rrs rrArgument.is_write& d*Gt/G/GGrcURnUR(a(US;deURSSURS35nURcU$SnUR(aSUR3nUSURU3$)NrpzTensor?zTensor[]rpTensor(r#r-rLr )rmrrr<r_)rrm mb_defaults rrArgument.__str__*s)) ??<< <<<<GDOO3DA*FGD 99 KJ|| / V1TYYK |4 4rrNr!rF)rrr!rLrr r"r#r$r%r&rrHrrJrrr'rrrrLrLs^ I J *"! $$LHH 5rrLcz\rSrSr%S\S'S\S'S\S'\SSj5r\SS j5r\SS j5r SS jr S r g )ri8rFr<rrmrFrcUR$r:rfrs rrHReturn.alias_info>rJrcSU;aURSS5upOUnSn[R"SU5nU(aUURS5S;dS5eSURS5-n[R URS55nOUnSn[ R U5n[UUUS9n[U5U:Xd[U5S U35eU$) Nr r5rMrlrNrOrprr) rRrrrrrJrrr)rrTr<rrVrrmrs rrJ Return.parseBs #:#&::c1#5 ND ND.? ;;q>& >>  >  A.F#))%++a.9J#FJzz&! !  1v}2QSE22}rcXURSL=(a URR$r:rrs rrReturn.is_writearZrcURnUR(a(US;deURSSURS35nURcU$USUR3$)Nr\rpr]r#r )rmrrr<)rrms rrReturn.__str__ese)) ??<< <<<<GDOO3DA*FGD 99 KV1TYYK( (rrNr`)rrr!rrr rarrrrr8sT  J!! <HH)rrc \rSrSr%S\S'Srg) SelfArgumentiqrLrvrN)r"r#r$r%r&r'rrrrlrlqsrrlcH\rSrSr%S\S'S\S'S\S'S\S'S SjrSrg ) TensorOptionsArgumentsiyrLdtypelayoutdevice pin_memoryc^URURURUR/$r:rorprqrrrs rallTensorOptionsArguments.alls! DKKdooFFrrNr!zSequence[Argument])r"r#r$r%r&rur'rrrrnrnys O  Grrnc\rSrSr%S\S'S\S'S\S'S\S'S\S 'S\S 'S\S '\S$S j5r\S$S j5r\S$Sj5r\S$Sj5r \S$Sj5r \S%Sj5r \S&Sj5r \S'Sj5r \S%Sj5rS(SjrS)SjrS)SjrS)SjrSS.S*SjjrS+SjrS,Sjr\S-Sj5r\S.Sj5rS/S jrS0S!jrS"rg#)1rDiztuple[Argument, ...]pre_self_positionalzSelfArgument | Nonerurpre_tensor_options_kwarg_onlyzTensorOptionsArguments | Nonerpost_tensor_options_kwarg_onlyrvcv/nURUR5 URUR5 U$r:)rrRrSrrks r flat_non_outArguments.flat_non_outs0  4''( 4''( rc/nURUR5 URb%URURR5 URUR 5 U$r:)rryruryrvrr}s rrRArguments.flat_positionalsP  4++, == $ JJt}}-- . 4,,- rcjURVs/sHoR(dMUPM sn$s snfr:)rrrrs rrt&Arguments.post_self_positional_mutables&44C4a 4CCCs00c/nURUR5 URb)URURR55 URUR5 U$r:)rrzrrur{r}s rrSArguments.flat_kwarg_onlysW  4556    * JJt**..0 1 4667 rc/nURUR5 URUR5 URUR5 U$r:)rrRrSrvr}s rrArguments.flat_alls@  4''( 4''( 488 rcv/nURUR5 URUR5 U$r:)r positional kwarg_onlyr}s rnon_outArguments.non_outs0GI 4??# 4??# rc/nURUR5 URbURUR5 URUR5 U$r:)rryruryrr}s rrArguments.positionalsJ-/ 4++, == $ JJt}} % 4,,- rc/nURUR5 URbURUR5 URUR5 U$r:)rrzrryr{r}s rrArguments.kwarg_onlysN79 4556    * JJt** + 4667 rc/nURUR5 URUR5 URUR5 U$r:)rrrrvr}s rru Arguments.alls<FH 4??# 4??# 488 rcURVs/sH;nURcMURR(dM/URPM= sn$s snfr:)rrrr<rs rmutable_arg_namesArguments.mutable_arg_namessL]] "|| ,-LL,A,A AFF"   sAAAc:[SUR55$)Nc3T# UHoRR5v M g7fr:)rmrrqs rr+Arguments.has_tensor_arg.. F4Eq66((**4E&(rr~rs rrArguments.has_tensor_argFD4E4EFFFrc:[SUR55$)Nc3T# UHoRR5v M g7fr:)rmrrqs rr+Arguments.has_symint_arg..rrrrs rrArguments.has_symint_argrrc:[SUR55$)Nc3T# UHoRR5v M g7fr:)rmrrqs rr.Arguments.has_generator_arg..s I7H!66++--7Hrrrs rrArguments.has_generator_argsIt7H7HIIIrFrc |^SU4Sjjn[[[X R55URb%[ U"URR 55OS[[X R55[[X R55[[X R55-SSSS9$)Ncp>[URURT(dURSS9$SSS9$)NrP)rLr<rmr_)rrs rstrip_arg_annotation1Arguments.signature..strip_arg_annotations=VVVV)6   =A  rrryrurrzrr{rv)rrLr!rL) rDrrryrurlrvrrzr{)rrrs ` rrArguments.signatures  %(*B*BC! ==,1$--2H2HIJ!&(*C*CD" +0(*L*LM+C,.Q.QRS+T  +-/  rc URce[R"U[[R"URRSS95S9$)Nrf)ru)ru dataclassesrrlrvrs rremove_self_annotation Arguments.remove_self_annotation sJ}}((("" !##DMM$:$:tL  rct[UR5S:Xde[R"U[ U5S9$)Nr)rv)rzrvrrr)rrs r with_out_argsArguments.with_out_args" s5488}!!!"" d   rcF/n/n/nUnURS5HnU(dM US:XaXALdS5eUnM![RU5nURb'URR(a XALaO XBLaUnOXCLdeUR U5 M XU4$)Nrrz:invalid syntax: kwarg-only specifier * can only occur once)rrLrJrrry)r\rrrv arguments_accrpargs r _preparseArguments._preparse) s%' %'  " ::d#Ccz$2P2!+ >>#&D*t/G/G ."0$'M$///   &/$2s**rc [RU5upnSn[U5HupVURS:XdMUn O UbUSUn[ X5nXS-Sn O/nSnUn /n Sn /n U n SSjnU"S[ R S55U"S[ R S55U"S [ R S 55U"S [ R S 55/nS nU[U5:aU[U5[U5- ::ac[S[XXU[U5-555(a5XLde[X%X%S-X%S-X%S-S9n U[U5- nU n MU RX%5 US- nU[U5:aM[[U5U[U 5[U 5U [U 5[U5S9$)z Input: 'int x, int y, int z' Nrr5c^^UU4Sj$)Ncd>URT:H=(a URT[T54;$r:)r<rmr)rr<tys r/Arguments.parse..pred..s s(QVVt^P2|B?O:P0PPrr)r<rs``rpredArguments.parse..predr s P PrrorrprrqrrrrBrc36# UHupU"U5v M g7fr:r)rprs rr"Arguments.parse.. s TaDD Tsrlrfrtr)r<rrrr!zCallable[[Argument], bool]) rDr enumerater<rlrrJrzrursrnryr)r\rrrvself_ixirryrurrzrr{kwarg_only_accr predicatess rrJArguments.parseM s '0&9&9$&?# j)DAvv*  ",Xg"6 #J$78H#-km#< "$ H#- 9;%8<9;&6 Q $**\2 3 4::h/ 0 4::h/ 0 tzz&1 2   #j/!C Oc*o55 #J13z??R0S T*JJJ%;(m)a%0)a%0#-!e#4 &N Z(A%CN  ! !*- 0 FA)#j/!, %&9 :!&';!<*/0M*N)+01O+Pc   rc/nUR[[UR55 UR(dUR (aUR S5 UR[[UR55 UR[[UR 55 SRU5$)Nrr)rrrrRrSrvryr)r all_argumentss rrArguments.__str__ s#% Sd&:&:;<   488   %Sd&:&:;<Sdhh/0yy''rcJURcUR(aeURcUR(aeURVs/sH1nURcMURR (dM/UPM3 nn[ U5S:XdS5egs snf)NrzOmutable pre_self_positional arguments are not currently supported in the schema)ruryrr{rrrz)rrmutable_pre_self_positionalss rrArguments.__post_init__ s == // //    &:: :: --( -|| ,-LL,A,A - %( /0A5 ] 5 ( sB $B B rNrw)r!z:Sequence[Argument | SelfArgument | TensorOptionsArguments])r!z!Sequence[Argument | SelfArgument])r!z+Sequence[Argument | TensorOptionsArguments]r!z list[str]r)rrBr!rD)r!rD)rlist[Argument]r!rD)r\rr!z5tuple[list[Argument], list[Argument], list[Argument]])r\rr!rDr r)r"r#r$r%r&rr~rRrtrSrrrrrurrrrrrrrrrJrrr'rrrrDrDsP .-!!..#7711%98   DD C GGJ27# J  !+!+FM M ^( rrD) rsubmuldivmodpowlshiftrshiftandxororcn\rSrSr%S\S'S\S'S\S'SrS\S'S rS \S '\SS j5rSS jr Sr g )ri rrrBrrFrNz Optional[str]r?c^ US:wdeURS5(aS5e[R"SU5nU(aURS5OSnU(aURS5OUn[R"SU5nUbGSnURS5m [ U 4S j[ 55(aSnT SSm O%S nT S S :wdeOS nUm T S S:XaSnT SS m OS nSnT RU5(a"SnT S[ U5*m U(dU(aeOS n[T UUUUS9n [U 5U:Xd[U 5SU35eU $)Nr-rzs_out suffix is reserved and not permitted for operator names; did you mean to specify an out overload name instead?z^(?:(.*)::)?(.*)$r5rlz ^__([^_]+)__$Tc34># UH nTSU3:Hv M g7f)rNr)rnrs rr)BaseOperatorName.parse.. sG,Fq4Qqc7?,FrFrrr"_ _functional)rrrrr?r) rrrrrAUGMENTED_ASSIGNMENT_NAMESrzrr) rrr? op_without_nsr8rrfunctional_suffixrrrs @rrJBaseOperatorName.parse s}Rxx;;v&&  D &-r2&+EKKN */ AR HH%} 5 = M771:DG,FGGGABx Aw#~%~!M DBx3CRy* ==* + +"& 130112D%W 44W"'  ' 3  1v|0AxtB400|rc.UR(aURS3OSnUR(a)UR(aSOSnUSUURS3$UR(aSOUR(aSOSnUURU3$)Nr`r-r__rr)r?rrrr)rnamespace_prefixrs rrBaseOperatorName.__str__+ s48NNdnn-R0   ||A&'r!TYYKr: :<<++# '' {1#6 6rr)rrr!rr ) r"r#r$r%r&rr?rrJrr'rrrrr sC I M(!&% $I}#66p 7rrcf\rSrSr%S\S'S\S'\S Sj5rSSjrSSjrSS jr SS jr S r g )rwi= rr<rrcSU;aURSS5upOUnSn[[RU5US9n[ U5U:Xd[ U5SU35eU$)N.r5r-r~r)rrwrrJr)r.r<rrs rrJOperatorName.parseB sk '>")--Q"7 D-DM .44T:- X1v :SVHD ":: rctUR(aURSUR3$UR$)Nrrr<rs rrOperatorName.__str__M 3   ii[$"4"4!56 6ii[ !rctUR(aURSUR3$UR$)Nrrrs runambiguous_nameOperatorName.unambiguous_nameX rrc[[URRSURRS9UR S9$NFrr~)rwrr<rrrrs rremove_inplaceOperatorName.remove_inplace^ s=!YY^^"ii55 ,,   rcx[[URRSURRS9US9$r)rwrr<rr)roverloads r with_overloadOperatorName.with_overloadh s7!YY^^"ii55 #   rrN)r.rr!rwr )r!rw)rrr!rw) r"r#r$r%r&rrJrrrrr'rrrrwrw= s4 ""   rrwcURR5[RL=(a8 UR U5(+=(a UR UR5$r:)rArrrr9)rr3bs r"gets_generated_out_inplace_wrapperrs sI  Z222 ' Q  ' LL &rcx\rSrSr%S\S'S\S'S\S'SSjrSS .SS jjr\SS j5r\SS j5r S r g)NativeFunctionsViewGroupi r>viewr view_copy view_inplacecJURR(deURcc[UR5(aH[ URR R 5S[UR5<S35eOURR R R RRS5(deURR R5URR RSS9:XdeSURR;dP[ URR R 5[ URR54S35eURbLURR R5URR R5:XdeURR(aURbvURR(d[[ URR R 5S[ URR R 5S 35eURR(aURbwURR(d[[ URR R 5S[ URR R 5S 35eggg) Nz appears to be a new operator that aliases its inputs. The codegen expects you to add a corresponding operator to native_functions.yaml: z3. See Note [view_copy NativeFunctions] for details.)rrTrrz appears to be a view_copy operator. The codegen expects view_copy operators to be annotated with the 'view_copy' tag in native_functions.yaml. See Note [view_copy NativeFunction] for details.rz] must either both have CompositeImplicitAutograd kernels, or both not have composite kernels.zi must either both have CompositeImplicitAutogradNestedTensor kernels, or both not have composite kernels.)rrrgets_generated_view_copyrrAr<get_view_copy_namerrrrZrrVrWrs rr&NativeFunctionsViewGroup.__post_init__ syy#### >> !/ :: tyy~~**+,-&tyy145EE ::>>&&++0055>>?TUU UU99>>++-1D1D1N1N%)2O2 $.."5"55 t~~**//0#diinn2EEFGDD 5    (99>>++-1B1B1G1G1Q1Q1SS SS 99 ; ;  ,((OO499>>../0c$:K:K:P:P:U:U6V5WXhhO 99 I I  ,((]]499>>../0c$:K:K:P:P:U:U6V5WXtt]- JrT) include_copyc## URv URbURv URbU(aURv ggg7fr:)rrr)rr s rr"NativeFunctionsViewGroup.functions sDii    (## # >> %,.. +7 %sAAc.URR$r:)rrrs rr"NativeFunctionsViewGroup.root_name syy"""rc.URR$r:)rrVrs r composite"NativeFunctionsViewGroup.composite syy???rrNr)r rBr!rr r) r"r#r$r%r&rrrrrr'rrrrr sP %$''!F15!##@@rrcUR(dgUR(agSUR;agURRRR R S5(agg)NFr_inverseT)rrVrZrAr<rr)rs rrr sV << //  vv{{%%j11 rcS/n[URR5U;a[U5(deURRRRS3n[ [ USURRRRS9URRRS9nU$)NnarrowrFrr~) rrAr<rrrwrrr)r-list_of_ops_with_explicit_view_copy_operatorsrview_copy_names rrr s6>J1 166;;LL'****66;;##(()/I! E9I9I9W9W ffkk// N rcUS:XagUSS:XaUSS:XaUSSn[SURS 555$) z Input: '()' Output: [] z()rrr!r"r#r5c3L# UHn[RU5v M g7fr:)rrJrs rr parse_returns.. sF.Esc"".Es"$r)rr)r]s rrYrY sN d1~RC!7!!B' Fk.?.?.EF FFrcR\rSrSr%S\S'S\S'\S Sj5rS SjrS SjrS r g )rxi zdict[str, list[Argument]]rrrct[U[5(de/nSUS;aLUSRS5nUVs/sH&n[R UR 55PM( nnUSSn0nUHn[U[ 5(deSU;dS5eURS5upgSU;d SUS35eURS5nUVs/sH&n[R UR 55PM( n nXU'M [XAS9n U R5U:XdS 5eU $s snfs snf) Nrr"rezLprecomputed parameters without replacement are allowed only in the last liner zillegal kernel param name 'z' in precomputed parameters')rrzr.to_list() != src) rorsrrLrJr}rrxto_list) srcadd_argsadd_listr<rraw_replace_itemr with_list_raw with_listwith_list_argsrs rrJPrecompute.parse sO#t$$$$ R 2w}}S)HAIJtzz|4HJcr(C # .44 44-- 5 - "2!7!7!? Cc> -cU2NO >&++C0IGPQythnnTZZ\:yNQ)CL!$ w 5yy{c!7#77!+K Rs -D0-D5c&URH-nURR5UR:waM-e URR 5H6nUH-nURR5UR:waM-e M8 gr:)rr<upperrr )rrr\s rrPrecompute.__post_init__2 skA66<<>QVV+ ++LL'')Dvv||~///*rc/nURR5H3up#SRSU55nURUSU35 M5 U$)Nrc38# UHn[U5v M g7fr:r)rparams rr%Precompute.to_list..> s$P=OESZZ=Orr)rrDrry)r replace_list kernel_paramreplacement_params replacementss rrPrecompute.to_list; sX 04 0B0B0D ,L99$P=O$PPL   <.\N C D1ErrN)robjectr!rxrr) r"r#r$r%r&rrJrrr'rrrrxrx s-'& ""H0rrxr )rr7r!rB)rr>r3rrr'r!rB)rr>r!rB)rr>r!rw)r]rr!rE)s __future__rrrPrrenumrrtypingrrr torchgen.utilsr r r collections.abcr rrr)rrrr AUTOGRAD_KEYSFRAGMENT_NAMESPACESr7rrrrhasattrrprintrr~rzryrrrrrrWrrrrrrrtrurvrwrrrrrrrrr^r~rrrrrdispatch_device_maprrr&rrrrrrrrrr'r4r9r>rrrrr'r@rrrornrrrrLrrlrnrDrrrwrrrrrYrx) components0rr>s" !44DD2: $***d (nssu((,>,,>yJ,>, #$:; _>$_>DD  B {BG,,+-A !H2$rd"pqrpst  ! OOOOOO  #''9OOOOOO))55))66    9 D*% ,V5H%P,,^46 05&  j' (8(8*:K:K'LM j%j667 i%Z0=3LL e!.u!5 i8P!P o* i 88"#;t;4d T$ $f Gf Gf GR $q q q h $(((, $   6 $888` $t@t@t@~  $111t $2"2"2"lT. $*t**( $(4((( $d@ $t* $W5W5W5t $4)4)4)p $ $GGG $r r r p * $e7e7e7T $2 2 2 j.3? * $;@;@;@|.0 G $;;;yU,s?T$