ї ÷╡еjё ЦСю≈UddlmZmZddlmZddlmZgd╒ZGd└dі╚Zeі╚Z ee d<Gd └d і╚ZGd└dі╚Zd e deefd└ZdS)И)зCallableзOptionalИ)зFakeImplHolder)зRegistrationHandle)зSimpleLibraryRegistryзSimpleOperatorEntryз singletoncС(≈eZdZdZd└Zdeddfd└ZdS)raJRegistry 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. cС≈i|_dS╘N╘з_data╘зselfs ЗOC:\Projects\Historical\venv\Lib\site-packages\torch/_library/simple_registry.pyз__init__zSimpleLibraryRegistry.__init__s ─ь┬▄ ┬ ┬ Сзqualnameзreturnr cС\≈||jvrt|і╚|j|<|j|Sr )rr ╘rrs rзfindzSimpleLibraryRegistry.finds0─ь≤4°:п%п%щ#6╟xя#@т#@┬D▄J░xя ь▄z≤(т#п#rN)з__name__з __module__з__qualname__з__doc__rзstrr╘rrrrsP─────П П ПППП$≤SП$п%:П$П$П$П$П$П$rrr cС4≈eZdZdZdefd└Zed└і╚ZdS)r z└This is 1:1 to an operator overload. The fields of SimpleOperatorEntry are Holders where kernels can be registered to. rcСd≈||_t|і╚|_t|і╚|_dSr )rrз fake_implзGenericTorchDispatchRuleHolderзtorch_dispatch_rulesrs rrzSimpleOperatorEntry.__init__.s1─ь%┬▄ щ)7╦я)Aт)A┬▄Е*╗8я4т4П т!п!п!rcС≈|jSr )r"rs rз abstract_implz!SimpleOperatorEntry.abstract_impl6s ─Ю▄~пrN)rrrrrrзpropertyr&rrrr r 'sT─────ППП ═П П П П ПППЯ└XПППrr cС.≈eZdZd└Zdededefd└Zd└ZdS)r#cС"≈i|_||_dSr )rrrs rrz'GenericTorchDispatchRuleHolder.__init__<s─ь┬▄ ь ┬▄ ┬ ┬ rзtorch_dispatch_classзfuncrcС═┤┤≈┴═┴і╚rt┴⌡d┴j⌡²і╚┌|┴j┴<┬┬fd└}t |і╚S)Nz8 already has a `__torch_dispatch__` rule registered for cС∙≈┴j┴=dSr r╘rr*s──rз deregisterz;GenericTorchDispatchRuleHolder.register..deregisterIsЬ─ь■ п/п0п0п0r)rзRuntimeErrorrrr)rr*r+r/s`` rзregisterz'GenericTorchDispatchRuleHolder.register@sЬЬ─П▐9┼9п)я*т*П щь'пpпpпaeтanпpпpЯТП П,0┬▄ п'я(П 1П 1П 1П 1П 1П 1У"═*я-т-п-rcС8≈|j═|dі╚Sr )rзgetr.s rrz#GenericTorchDispatchRuleHolder.findNs─ь▄z▐~┼~п2╟Dя9т9п9rN) rrrrзtyperrr1rrrrr#r#;s^─────П!П!П!П.ь$(П.ь08П.Ю П.П.П.П.П:П:П:П:П:rr#r*rcСp≈t═|jі╚j═|і╚Sr )r rrr$)зopr*s rзfind_torch_dispatch_ruler7Rs.─щ▐>┼>≤"°/я*т*т?вDрDьЯТПrN)зtypingrrr"rзutilsrз__all__rr з__annotations__r r#r4r7rrrЗr<sПЮ%п%п%п%п%п%п%п%п%Ю%п%п%п%п%п%ь%п%п%п%п%п%ПHп Gп G─П$П$П$П$П$Я$Т$П$П2$9п#8я#:т#:─ п п:п:я:ПППППЯТПП(:П:П:П:П:Я:Т:П:П.╟tПюхт@RППППППr