U ʹje @sUddlZddlZddlmZmZmZmZmZmZm Z ddl Z ddl m Z GdddeZ ee j ee jge fZeZeeefed<deeeeeedd d Zejed d Zejed d ZddZdeedddZedddZededddZdS)N)CallableDictListOptionalProtocolSequenceTuple)fxc@s(eZdZejeejdfdddZdS) CompiledFn.)argsreturncGsdS)N)selfr r r ^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_dynamo/backends/registry.py__call__ szCompiledFn.__call__N)__name__ __module__ __qualname__torchTensorrrr r r rr sr _BACKENDSr  compiler_fnnametagscCsZ|dkrtjt||dSt|s$t|p,|j}|tksDtd||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_backendcallableAssertionErrorrrtuple_tagsrr r rrs   r)debug)r) experimentalcCsRt|trN|tkrt|tkr(t||tkrFddlm}||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)setsortedritemsr1r r1r list_backendsCs  r6cCs<ddlm}ddlm}||ddlm}|dk s8tdS)Nr$)backends)import_submodule)dynamo_minifier_backend)r7utilsr8Zrepro.after_dynamor9r)r7r8r9r r rr)Ts    r) backend_namecsddlm}d}d}tjdkrX|}fdd||dD}t|dkrx|d}n ||d}|jkrx|}|dk rtt krt |d dS) Nr) entry_pointsZtorch_dynamo_backends) csg|]}|jkr|qSr r&)r/epr<r rr2hs z,_lazy_import_entry_point..r )group)rr) importlib.metadatar>sys version_infogetlenloadnamesr6r r)r=r>r group_nameZ backend_epsepsr r<rr*`s      r*)NNr )r-)rrDtypingrrrrrrrrr r Z GraphModulerZ CompilerFndictrr(__annotations__rrZregister_debug_backendZregister_experimental_backendr,r6 lru_cacher)r*r r r rs6$