9i "%SrSSKrSSKrSSKrSSKJr SSKJr SSKJ r J r J r J r J r SSKrSSKJr \R "\5r"SS\ 5r\ \R(\\R,/\4r0r\\\ \4\S '0r\\\4\S 'SS \ \S \ \S \\S\ S\ 44Sjjr\R<"\SS9r\R<"\SS9r S \ \\4S\4Sjr!SS\\4Sjjr"\RFSSj5r$\RFSSj5r%g)a This module implements TorchDynamo's backend registry system for managing compiler backends. The registry provides a centralized way to register, discover and manage different compiler backends that can be used with torch.compile(). It handles: - Backend registration and discovery through decorators and entry points - Lazy loading of backend implementations - Lookup and validation of backend names - Categorization of backends using tags (debug, experimental, etc.) Key components: - CompilerFn: Type for backend compiler functions that transform FX graphs - _BACKENDS: Registry mapping backend names to entry points - _COMPILER_FNS: Registry mapping backend names to loaded compiler functions Example usage: @register_backend def my_compiler(fx_graph, example_inputs): # Transform FX graph into optimized implementation return compiled_fn # Use registered backend torch.compile(model, backend="my_compiler") The registry also supports discovering backends through setuptools entry points in the "torch_dynamo_backends" group. Example: ``` setup.py --- from setuptools import setup setup( name='my_torch_backend', version='0.1', packages=['my_torch_backend'], entry_points={ 'torch_dynamo_backends': [ # name = path to entry point of backend implementation 'my_compiler = my_torch_backend.compiler:my_compiler_function', ], }, ) ``` ``` my_torch_backend/compiler.py --- def my_compiler_function(fx_graph, example_inputs): # Transform FX graph into optimized implementation return compiled_fn ``` Using `my_compiler` backend: ``` import torch model = ... # Your PyTorch model optimized_model = torch.compile(model, backend="my_compiler") ``` N)Sequence) EntryPoint)AnyCallableOptionalProtocolUnion)fxcX\rSrSrS\R S\\R S44SjrSrg) CompiledFnKargsreturn.cg)N)selfrs _/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/torch/_dynamo/backends/registry.py__call__CompiledFn.__call__LsrN) __name__ __module__ __qualname____firstlineno__torchTensortupler__static_attributes__rrrr r Ks!LellLuU\\35F/GLrr _BACKENDS _COMPILER_FNS compiler_fnnametagsr.c Uc[R"[XS9$[U5(deU=(d URnU[ ;d SU35eU[ ;a S[ U'U[ U'[U5UlU$)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)r"r#zduplicate name: ) functoolspartialregister_backendcallablerr rr_tags)r!r"r#s rr'r'Us   !1HH K   ';''D } $?(8&?? $)# $%M$d K r)debug)r#) experimentalc[U[5(aaU[;a [5 U[;a SSKJn U"US9eU[ ;a$[UnUb[UR5US9 [ UnU$)z#Expand backend strings to functions)InvalidBackend)r")r!r") isinstancestrr _lazy_importexcr.r r'load)r!r. entry_points rlookup_backendr5xsm+s## i ' N i ' , k2 2 m +#K0K& [-=-=-?kR#K0 rc [5 [U=(d S5n[R5Vs/sH8nU[;d)UR [UR 5(aM6UPM: nn[U5$s snf)zU Return valid strings that can be passed to: torch.compile(..., backend="name") r)r1setrkeysr intersectionr)sorted) exclude_tagsexclude_tags_setr"backendss r list_backendsr>su N<-2.NN$$D } $,,]4-@-F-FG $  (  s 5B.BcVSSKJn SSKJn U"U5 SSKJn Uce[ 5 g)Nr-)r=)import_submodule)dynamo_minifier_backend)r=utilsr@repro.after_dynamorA_discover_entrypoint_backends)r=r@rAs rr1r1s%(X< " .. .!#rcSSKJn Sn[RS:a/U"5nX;aX!O/nUVs0sHo3RU_M nnO$U"US9nUR Vs0sHoUX%_M nnUHnXF[ U'M gs snfs snf)Nr) entry_pointstorch_dynamo_backends) )group)importlib.metadatarGsys version_infor"namesr)rG group_nameepsepeps_dictr" backend_names rrErEs0(J '!n!+!2co*-.#BGGRK#.,03 : #)O : "*"8 ,! /;s B B )NNr))r*r+)rN)&__doc__r%loggingrMcollections.abcrrLrtypingrrrrr rr getLoggerrlogr GraphModulelistr CompilerFnrdictr0__annotations__r r'r&register_debug_backendregister_experimental_backendr5r>cacher1rErrrrcsj:x $);; !MMr~~tELL'9:JF G -/ 4Xj)) */') tCO$))-*% 3- 3-c3h :#**+;*M ) 1 1,! c:o 6:&T#Y$  $ $  9 9r