o ߗid@sFUddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z ddl mZeeZGddde Zeejee jgefZiZeee efed<iZeeefed<   dd e ed e ed e efd dZejeddZejeddZddZddeefddZe dddZ!e dddZ"dS)N) EntryPoint)CallableDictListOptionalProtocolSequenceTuple)fxc@s*eZdZdejdeejdffddZdS) CompiledFnargsreturn.cGsdS)N)selfr rr]/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/_dynamo/backends/registry.py__call__szCompiledFn.__call__N)__name__ __module__ __qualname__torchTensorr rrrrrr s"r _BACKENDS _COMPILER_FNSr compiler_fnnametagscCsj|dur tjt||dSt|sJ|p|j}|tvs"Jd||tvr*dt|<|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_backendcallablerrrtuple_tags)rrrrrrrs   r)debug)r) experimentalcCsbt|tr/|tvr t|tvrddlm}||d|tvr+t|}t||dt|}|S)z#Expand backend strings to functions)InvalidBackendr)rr) isinstancestrr _lazy_importexcr%rrload)rr% entry_pointrrrlookup_backend>s   r-r"r#r cs0ttpdfddtD}t|S)za Return valid strings that can be passed to: torch.compile(..., backend="name") rcs(g|]}|tvst|js|qSr)r intersectionr!.0r exclude_tagsrr Xsz!list_backends..)r)setrkeyssorted)r3backendsrr2r list_backendsOs   r9cCsBddlm}ddlm}||ddlm}|dusJtdS)Nr$)r8)import_submodule)dynamo_minifier_backend)r8utilsr:repro.after_dynamor;_discover_entrypoint_backends)r8r:r;rrrr)as     r)cs|ddlm}d}tjdkr"||vr|ngddDn||dfddjDD]}|t|<q3dS) Nr) entry_pointstorch_dynamo_backends) cSsi|]}|j|qSrr&)r1eprrr xsz1_discover_entrypoint_backends..)groupcsi|]}||qSrrr0epsrrrE{s)importlib.metadatar@sys version_infonamesr)r@ group_name backend_namerrGrr?os   r?)NNr)r.)#rloggingrJrIrtypingrrrrrrr rr getLoggerrlogr GraphModuler CompilerFnrr(__annotations__rrrregister_debug_backendregister_experimental_backendr-r9 lru_cacher)r?rrrrs@  $