9i UddlmZddlmZmZddlmZddlmZddl m Z gdZ Gdd Z e Z e ed <Gd d ZGd dZdededeedeffdZdS))Callable)AnyOptional) EffectHolder)FakeImplHolder)RegistrationHandle)SimpleLibraryRegistrySimpleOperatorEntry singletonc*eZdZdZddZdeddfdZdS) 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|_dSN_dataselfs Q/var/www/icac/venv/lib/python3.11/site-packages/torch/_library/simple_registry.py__init__zSimpleLibraryRegistry.__init__s 57 qualnamer cr|j|d}|t|x|j|<}|Sr)rgetr )rrress rfindzSimpleLibraryRegistry.finds:jnnXt,, ;)eZdZdZdeddfdZedefdZdS)r zThis is 1:1 to an operator overload. The fields of SimpleOperatorEntry are Holders where kernels can be registered to. rrNc||_t||_t||_t ||_dSr)rr fake_implGenericTorchDispatchRuleHoldertorch_dispatch_rulesreffectrrs rrzSimpleOperatorEntry.__init__0s@% )7)A)A *8 4 4 !%1$:$: rc|jSr)r&rs r abstract_implz!SimpleOperatorEntry.abstract_impl:s ~r) rrr r!r"rpropertyrr,r#rrr r )sg ;;;;;;~Xrr cneZdZdeddfdZdededefdefdZ dede edeffd Z dS) r'rrNc"i|_||_dSr)rrr*s rrz'GenericTorchDispatchRuleHolder.__init__@s57 % rtorch_dispatch_classfunc.crtdj|j<dfd }t |S)Nz8 already has a `__torch_dispatch__` rule registered for rcj=dSrrrr0sr deregisterz;GenericTorchDispatchRuleHolder.register..deregisterMs /000rr)r RuntimeErrorrrr )rr0r1r5s`` rregisterz'GenericTorchDispatchRuleHolder.registerDs 99) * * 'ppaeanpp ,0 '( 1 1 1 1 1 1 1"*---rc8|j|dSr)rrr4s rrz#GenericTorchDispatchRuleHolder.findRsz~~2D999r) rrr r"rtyperrr r7rrr#rrr'r'?s&&&&&& .$( .08c0B .  . . . .::(8CH;M2N::::::rr'opr0r.cpt|jj|Sr)r rr r()r:r0s rfind_torch_dispatch_ruler<Vs0 >>"/ * * ? D D  rN)collections.abcrtypingrreffectsrr&rutilsr __all__r r __annotations__r r'r9r<r#rrrCsN$$$$$$$ !!!!!!%%%%%%%%%%%% H G G4$9#8#:#: :::,::::::::. #' hsCx !r