Ë L²juãóÂ—UddlmZddlmZddlmZddlmZddlm Z gd¢Z Gd„d «Ze«Zee d <Gd„d«ZGd „d«Zdedededefdzfd„Zy)é)ÚCallable)ÚAnyé)ÚEffectHolder)ÚFakeImplHolder)ÚRegistrationHandle)ÚSimpleLibraryRegistryÚSimpleOperatorEntryÚ singletoncó(—eZdZdZdd„Zdeddfd„Zy) r aJRegistry for the "simple" torch.library APIs The "simple" torch.library APIs are a higher-level API on top of the raw PyTorch DispatchKey registration APIs that includes: - fake impl Registrations for these APIs do not go into the PyTorch dispatcher's table because they may not directly involve a DispatchKey. For example, the fake impl is a Python function that gets invoked by FakeTensor. Instead, we manage them here. SimpleLibraryRegistry is a mapping from a fully qualified operator name (including the overload) to SimpleOperatorEntry. ÚreturnNcó—i|_y©N©Ú_data©Úselfs ú]/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/_library/simple_registry.pyÚ__init__zSimpleLibraryRegistry.__init__s €Ø57ˆ óÚqualnamer cóv—|jj|d«}|€t|«x|j|<}|Sr)rÚgetr )rrÚress rÚfindzSimpleLibraryRegistry.finds8€Øj‰jn‰n˜X tÓ,ˆØˆ;Ü)<¸XÓ)FÐFˆDJ‰JxÑ 3Øˆ r©r N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rÚstrr©rrr r s „ñ ó8ð˜SðÐ%:ôrr rcó6—eZdZdZdeddfd„Zedefd„«Zy)r z„This is 1:1 to an operator overload. The fields of SimpleOperatorEntry are Holders where kernels can be registered to. rr Ncór—||_t|«|_t|«|_t|«|_yr)rrÚ fake_implÚGenericTorchDispatchRuleHolderÚtorch_dispatch_rulesrÚeffect©rrs rrzSimpleOperatorEntry.__init__0s3€Ø%ˆŒ Ü)7¸Ó)AˆŒä*¨8Ó4ð Ô!ô%1°Ó$:ˆrcó—|jSr)r%rs rÚ abstract_implz!SimpleOperatorEntry.abstract_impl:s€à~‰~Ðr) rrrr r!rÚpropertyrr+r"rrr r )s5„ñð; ð;¨ó;ðð˜~òóñrr cóZ—eZdZdeddfd„Zdededefdefd„Z dededefdzfd „Z y) r&rr Ncó —i|_||_yr)rrr)s rrz'GenericTorchDispatchRuleHolder.__init__@s€Ø57ˆŒ Ø%ˆ rÚtorch_dispatch_classÚfunc.cóž‡‡—‰j‰«rt‰›d‰j›«‚|‰j‰<dˆˆfd„}t |«S)Nz8 already has a `__torch_dispatch__` rule registered for có •—‰j‰=yrr©rr/s€€rÚ deregisterz;GenericTorchDispatchRuleHolder.register..deregisterMsø€Ø— ‘ Ð/Ñ0rr)rÚRuntimeErrorrrr)rr/r0r4s`` rÚregisterz'GenericTorchDispatchRuleHolder.registerDsZù€ð9‰9Ð)Ô*ÜØ'Ð(Ð(`Ðae×anÑanÐ`oÐpóð ð,0ˆ ‰ Ð'Ñ(ö 1ô" *Ó-Ð-rcó:—|jj|d«Sr)rrr3s rrz#GenericTorchDispatchRuleHolder.findRs€Øz‰z~‰~Ð2°DÓ9Ð9r)rrrr!rÚtyperrrr6rr"rrr&r&?s\„ð& ð&¨ó&ð.Ø$(ð.Ø08¸¸c¸Ñ0Bð.à ó.ð:¨ð:°(¸3À¸8Ñ2DÀtÑ2Kô:rr&Úopr/r .Ncór—tj|j«jj|«Sr)rrrr')r9r/s rÚfind_torch_dispatch_ruler;Vs-€ô>‰>˜"Ÿ/™/Ó*×?Ñ?×DÑDØóðr)Úcollections.abcrÚtypingrÚeffectsrr%rÚutilsrÚ__all__r rÚ__annotations__r r&r8r;r"rrúrBsuðÞ$Ýå!Ý%Ý%òH€÷ññ4$9Ó#:€ Ð Ó:÷ñ÷,:ñ:ð.ØðØ#'ðà ˆc3ˆhÑ˜$Ñôr