o I&i @sUddlZddlZddlmZmZmZmZmZmZm Z ddl Z ddl m Z GdddeZ ee j ee jge fZeZeeefed<   ddeed eed eefd d Zejed dZejeddZddZddeefddZedddZeddefddZdS)N)CallableDictListOptionalProtocolSequenceTuple)fxc@s*eZdZdejdeejdffddZdS) CompiledFnargsreturn.cGsdS)N)selfr r r JC:\wamp64\www\opt\env\Lib\site-packages\torch/_dynamo/backends/registry.py__call__ szCompiledFn.__call__N)__name__ __module__ __qualname__torchTensorrrr r r rr s"r _BACKENDSr compiler_fnnametagscCsZ|dur tjt||dSt|sJ|p|j}|tvs"Jd||t|<t||_|S)a Decorator to add a given compiler to the registry to allow calling `torch.compile` with string shorthand. Note: for projects not imported by default, it might be easier to pass a function directly as a backend and not use a string. Args: compiler_fn: Callable taking a FX graph and fake tensor inputs name: Optional name, defaults to `compiler_fn.__name__` tags: Optional set of string tags to categorize backend with N)rrzduplicate name: ) functoolspartialregister_backendcallablerrtuple_tags)rrrr r rrs   r)debug)r) experimentalcCsRt|tr'|tvr t|tvrt||tvr#ddlm}||dt|}|S)z#Expand backend strings to functions)InvalidBackendr) isinstancestrr _lazy_import_lazy_import_entry_pointexcr#)rr#r r rlookup_backend4s   r*r r!r cs,ttpdtfddtDS)za Return valid strings that can be passed to: torch.compile(..., backend="name") r cs g|] \}}|js|qSr ) intersectionr).0rbackendZ exclude_tagsr r Ls  z!list_backends..)r'setsortedritemsr/r r/r list_backendsCs  r4cCs<ddlm}ddlm}||ddlm}|dusJdS)Nr")backends)import_submodule)dynamo_minifier_backend)r5utilsr6Zrepro.after_dynamor7)r5r6r7r r rr'Ts   r' backend_namecsddlm}d}d}tjdkr,|}fdd||dD}t|dkr+|d}n||d}|jvr<|}|durOtt vrQt |d dSdSdS) Nr) entry_pointsZtorch_dynamo_backends) csg|] }|jkr|qSr r$)r-epr:r rr0hsz,_lazy_import_entry_point..r )group)rr) importlib.metadatar;sys version_infogetlenloadnamesr4rr)r:r;r group_nameZ backend_epsZepsr r?rr(`s       r()NNr )r+)rrBtypingrrrrrrrrr r Z GraphModulerZ CompilerFndictrr&__annotations__rrZregister_debug_backendZregister_experimental_backendr*r4 lru_cacher'r(r r r rs8 $