i UddlmZddlmZmZddlmZddlmZddl m Z gdZ Gdd Z e Z e ed <Gd d ZGd dZdededeedeffdZy))Callable)AnyOptional) EffectHolder)FakeImplHolder)RegistrationHandle)SimpleLibraryRegistrySimpleOperatorEntry singletonc(eZdZdZddZdeddfdZy) 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. returnNci|_yN_dataselfs X/var/www/html/engine/venv/lib/python3.12/site-packages/torch/_library/simple_registry.py__init__zSimpleLibraryRegistry.__init__s 57 qualnamer cv|jj|d}|t|x|j|<}|Sr)rgetr )rrress rfindzSimpleLibraryRegistry.finds8jjnnXt, ;).deregisterMs /0rr)r RuntimeErrorrrr )rr0r1r5s`` rregisterz'GenericTorchDispatchRuleHolder.registerDsZ 99) *'((`aeanan`op ,0 '( 1"*--rc:|jj|dSr)rrr4s rrz#GenericTorchDispatchRuleHolder.findRszz~~2D99r) rrr r"rtyperrr r7rrr#rrr'r'?s\&&& .$( .08c0B .  .::(8CH;M2N:rr'opr0r.crtj|jjj|Sr)r rr r()r:r0s rfind_torch_dispatch_ruler<Vs- >>"// * ? ? D D rN)collections.abcrtypingrreffectsrr&rutilsr __all__r r __annotations__r r'r9r<r#rrrCsu$ !%% H4$9#: :,::. #' hsCx !r