ifddlmZmZmZddlmZddlmZddlm Z Gdde Z de de d d fd Z y ) )AnyCallableOptional)override)MisconfigurationException)_register_classesc eZdZdZ ddedeededededef d Z e dded eedefd Z deddfd Z de efd ZdefdZy)_AcceleratorRegistryadThis class is a Registry that stores information about the Accelerators. The Accelerators are mapped to strings. These strings are names that identify an accelerator, e.g., "gpu". It also returns Optional description and parameters to initialize the Accelerator, which were defined during the registration. The motivation for having a AcceleratorRegistry is to make it convenient for the Users to try different accelerators by passing mapped aliases to the accelerator flag to the Trainer. Example:: @AcceleratorRegistry.register("sota", description="Custom sota accelerator", a=1, b=True) class SOTAAccelerator(Accelerator): def __init__(self, a, b): ... or AcceleratorRegistry.register("sota", SOTAAccelerator, description="Custom sota accelerator", a=1, b=True) Nname accelerator descriptionr init_paramsreturnc ttstdvr|stddi|d<|d<dtdtffd }|||S|S) aRegisters a accelerator mapped to a name and with required metadata. Args: name : the name that identifies a accelerator, e.g. "gpu" accelerator : accelerator class description : accelerator description override : overrides the registered accelerator, if True init_params: parameters to initialize the accelerator z`name` must be a str, found 'z@' is already present in the registry. HINT: Use `override=True`.r rr rc&|d<d<<|S)Nr accelerator_name)r datar selfs {/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_fabric/accelerators/registry.py do_registerz2_AcceleratorRegistry.register..do_registerLs'"-D '+D# $DJ ) isinstancestr TypeErrorrr)rr r r rrrrs`` @rregisterz_AcceleratorRegistry.register/s$ 4 5:4&AB B 4<+av5u,vw w!)])] X (   "{+ +rdefaultc||vr||}|ddi|dS||Sd}|j}t|j||)zCalls the registered accelerator with the required parameters and returns the accelerator object. Args: name (str): the name that identifies a accelerator, e.g. "gpu" r rz/'{}' not found in registry. Available names: {}r)available_acceleratorsKeyErrorformat)rr rrerr_msgavailable_namess rgetz_AcceleratorRegistry.getWsb 4<:D&4 &=m)<= =  NC557w~~dO<==rc&|j|y)z+Removes the registered accelerator by name.N)pop)rr s rremovez_AcceleratorRegistry.removejs  rc4t|jS)z)Returns a set of registered accelerators.)setkeysrs rr z+_AcceleratorRegistry.available_acceleratorsns499;rc^djdj|jS)NzRegistered Accelerators: {}z, )r"joinr r,s r__str__z_AcceleratorRegistry.__str__rs%,33DIId>Y>Y>[4\]]r)NF)N)__name__ __module__ __qualname____doc__rrrboolrrrr%r(r*r r/rrrr r s6+/ &&h'& &  &  & &P>>hsm>s>>$34 C ^^rr registry base_modulerNcVddl}|j|}ddlm}t |d||y)zLegacy. Do not use. rN) Acceleratorregister_accelerators) importlib import_module)lightning_fabric.accelerators.acceleratorr9r)r6r7r;moduler9s rcall_register_acceleratorsr?vs)   $ $[ 1FEh 7Mr)typingrrrtyping_extensionsr%lightning_fabric.utilities.exceptionsr#lightning_fabric.utilities.registryrdictr rr?rrrrEsG+*&KA]^4]^@ N)= NC NTX Nr