o z“©ióã@sRddlmZddlmZmZddlZddlmZ ddl m Z Gdd„de eƒZdS)é)ÚABC)ÚAnyÚDictN)Ú Accelerator)Ú_DEVICEc@s4eZdZdZd dd„Zdedeeeffd d „Z dS) rz¨The Accelerator base class for Lightning PyTorch. .. warning:: Writing your own accelerator is an :ref:`experimental ` feature. Útrainerú pl.TrainerÚreturnNcCsdS)z¡Called by the Trainer to set up the accelerator before the model starts running on the device. Args: trainer: the trainer instance N©)Úselfrr r ú^/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_lightning/accelerators/accelerator.pyÚsetupszAccelerator.setupÚdevicecCst‚)z¡Get stats for a given device. Args: device: device for which to get stats Returns: Dictionary of device stats )ÚNotImplementedError)r rr r r Úget_device_stats%s zAccelerator.get_device_stats)rrr N) Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrÚstrrrr r r r rs r) ÚabcrÚtypingrrÚpytorch_lightningÚplÚ)lightning_fabric.accelerators.acceleratorrÚ _AcceleratorÚ lightning_fabric.utilities.typesrr r r r Ús