a d @sUddlZddlZddlmZmZmZmZmZmZm Z ddl Z ddl m Z GdddeZ ee j ee jge fZeZeeefed<deeeeeedd d Zejed d Zejed d ZddZdddZedddZededddZdS)N)CallableDictListOptionalProtocolSequenceTuple)fxc@s(eZdZejeejdfdddZdS) CompiledFn.)argsreturncGsdS)N)selfr r r h/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_dynamo/backends/registry.py__call__ szCompiledFn.__call__N)__name__ __module__ __qualname__torchTensorrrr r r rr sr _BACKENDSr  compiler_fnnametagscCsZ|durtjt||dSt|s$J|p,|j}|tvsDJd||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_tagsrr r rrs   r)debug)r) experimentalcCs4t|tr0|tvrt|tvr(t|t|}|S)z#Expand backend strings to functions) isinstancestrr _lazy_import_lazy_import_entry_point)rr r rlookup_backend4s 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 Hs z!list_backends..)r%setsortedritemsr,r r,r list_backends?s  r2cCs<ddlm}ddlm}||ddlm}|dus8JdS)N)backends)import_submodule)dynamo_minifier_backend)r4utilsr5Z debug_utilsr6)r4r5r6r r rr%Ps    r% backend_namecsddlm}d}d}tjdkrT|}fdd||D}t|dkrt|d}n ||d}|jvrt|}|durttvrt |ddS) Nr) entry_pointsZtorch_dynamo_backends) csg|]}|jkr|qSr )r)r*epr9r rr.dz,_lazy_import_entry_point..)group)rr) importlib.metadatar;sys version_infolenloadnamesr2rr)r:r;rZ group_nameZ backend_epsZepsr r9rr&\s      r&)NNr )r()rrBtypingrrrrrrrrr r Z GraphModulerZ CompilerFndictrr$__annotations__rrZregister_debug_backendZregister_experimental_backendr'r2 lru_cacher%r&r r r rs2 $