DŽgN UddlmZmZddlmZddlmZgdZGddZeZ ee d<Gd d Z Gd d Z d e deefdZy))CallableOptional)FakeImplHolder)RegistrationHandle)SimpleLibraryRegistrySimpleOperatorEntry singletonc&eZdZdZdZdeddfdZy)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. ci|_yN_dataselfs f/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/_library/simple_registry.py__init__zSimpleLibraryRegistry.__init__s  qualnamereturnr cl||jvrt||j|<|j|Sr )rr rrs rfindzSimpleLibraryRegistry.finds0 4:: %#6x#@DJJx zz(##rN)__name__ __module__ __qualname____doc__rstrrrrrr s  $S$%:$rrr c,eZdZdZdefdZedZy)r zThis is 1:1 to an operator overload. The fields of SimpleOperatorEntry are Holders where kernels can be registered to. rcR||_t||_t||_yr )rr fake_implGenericTorchDispatchRuleHoldertorch_dispatch_rulesrs rrzSimpleOperatorEntry.__init__.s%% )7)A *8 4 !rc|jSr )r"rs r abstract_implz!SimpleOperatorEntry.abstract_impl6s ~~rN)rrrrrrpropertyr&rrrr r 's'   rr c,eZdZdZdededefdZdZy)r#c i|_||_yr )rrrs rrz'GenericTorchDispatchRuleHolder.__init__<s   rtorch_dispatch_classfuncrcjrtdj|j<fd}t |S)Nz8 already has a `__torch_dispatch__` rule registered for c j=yr rrr*sr deregisterz;GenericTorchDispatchRuleHolder.register..deregisterIs /0r)r RuntimeErrorrrr)rr*r+r/s`` rregisterz'GenericTorchDispatchRuleHolder.register@sZ 99) *'((`aeanan`op ,0 '( 1"*--rc:|jj|dSr )rgetr.s rrz#GenericTorchDispatchRuleHolder.findNszz~~2D99rN) rrrrtyperrr1rrrrr#r#;s)! .$( .08 .  .:rr#r*rcrtj|jjj|Sr )r rrr$)opr*s rfind_torch_dispatch_ruler7Rs+ >>"// * ? ? D D rN)typingrrr"rutilsr__all__rr __annotations__r r#r4r7rrrr<s_%%%% H$$2$9#: :(::.t@Rr