o he @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 U/var/www/html/ai/venv/lib/python3.10/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).0rbackend exclude_tagsr r Ls  z!list_backends..)r'setsortedritemsr/r r/r list_backendsCs  r5cCs<ddlm}ddlm}||ddlm}|dusJdS)Nr")backends)import_submodule)dynamo_minifier_backend)r6utilsr7repro.after_dynamor8)r6r7r8r r rr'Ts   r' backend_namecsddlm}d}d}tjdkr,|}fdd||dD}t|dkr+|d}n||d}|jvr<|}|durOtt vrQt |d dSdSdS) Nr) entry_pointstorch_dynamo_backends) csg|] }|jkr|qSr r$)r-epr<r rr1hsz,_lazy_import_entry_point..r )group)rr) importlib.metadatar=sys version_infogetlenloadnamesr5rr)r<r=r group_name backend_epsepsr rBrr(`s       r()NNr )r+)rrEtypingrrrrrrrrr r GraphModuler CompilerFndictrr&__annotations__rrregister_debug_backendregister_experimental_backendr*r5 lru_cacher'r(r r r rs8 $