Ë
      çi  ã                   ó@  — d dl Z d dlmZ d dlZd dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ dd„Z G d	„ d
e«      Z G d„ de«      Z G d„ de«      Z G d„ de«      Z G d„ de«      Zdd„Z e«         e«        ej-                  ej.                  j0                  «       y)é    N)ÚAny)ÚXLAAccelerator)ÚXLAPrecision)Ú_StrategyRegistry)ÚSingleDeviceXLAStrategy)Úrank_zero_deprecationc                  óð   — t         j                  t           } | t         j                  d<   | t         j                  d<   | t         j                  d<   | t         j                  d<   | t         j                  d<   y )Nz&lightning_fabric.strategies.single_tpuz!lightning_fabric.accelerators.tpuz&lightning_fabric.plugins.precision.tpuz+lightning_fabric.plugins.precision.tpu_bf16z*lightning_fabric.plugins.precision.xlabf16)ÚsysÚmodulesÚ__name__)Úselfs    út/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/lightning_fabric/_graveyard/tpu.pyÚ_patch_sys_modulesr      s[   € Ü�;‰;”xÑ €DØ<@„C‡K�KÐ8Ñ9Ø7;„C‡K�KÐ3Ñ4Ø<@„C‡K�KÐ8Ñ9ØAE„C‡K�KÐ=Ñ>Ø@D„C‡K�KÐ<Ò=ó    c                   óJ   ‡ — e Zd ZdZdededdfˆ fd„Zededdfd„«       Zˆ xZ	S )	ÚSingleTPUStrategyznLegacy class.

    Use :class:`~lightning_fabric.strategies.single_xla.SingleDeviceXLAStrategy` instead.

    ÚargsÚkwargsÚreturnNc                 ó:   •— t        d«       t        ‰| �  |i |¤Ž y )NzBThe 'single_tpu' strategy is deprecated. Use 'single_xla' instead.©r   ÚsuperÚ__init__©r   r   r   Ú	__class__s      €r   r   zSingleTPUStrategy.__init__)   s   ø€ ÜÐbÔcÜ‰Ñ˜$Ð) &Ó)r   Ústrategy_registryc                 ó6   — d|vr|j                  d| d¬«       y y )NÚ
single_tpuz'Legacy class. Use `single_xla` instead.)Údescription)Úregister)Úclsr   s     r   Úregister_strategiesz%SingleTPUStrategy.register_strategies-   s&   € àÐ0Ñ0Ø×&Ñ& |°SÐFoÐ&Õpð 1r   )
r   Ú
__module__Ú__qualname__Ú__doc__r   r   Úclassmethodr   r"   Ú__classcell__©r   s   @r   r   r   "   sI   ø„ ñð*˜cð *¨Sð *°Tõ *ð ðqÐ4Eð qÈ$ò qó ôqr   r   c                   ó0   ‡ — e Zd ZdZdededdfˆ fd„Zˆ xZS )ÚTPUAcceleratorz`Legacy class.

    Use :class:`~lightning_fabric.accelerators.xla.XLAAccelerator` instead.

    r   r   r   Nc                 ó:   •— t        d«       t        ‰| �  |i |¤Ž y )NzeThe `TPUAccelerator` class is deprecated. Use `lightning_fabric.accelerators.XLAAccelerator` instead.r   r   s      €r   r   zTPUAccelerator.__init__:   s"   ø€ ÜØsô	
ô 	‰Ñ˜$Ð) &Ó)r   ©r   r#   r$   r%   r   r   r'   r(   s   @r   r*   r*   3   ó'   ø„ ñð*˜cð *¨Sð *°T÷ *ñ *r   r*   c                   ó0   ‡ — e Zd ZdZdededdfˆ fd„Zˆ xZS )ÚTPUPrecisionúcLegacy class.

    Use :class:`~lightning_fabric.plugins.precision.xla.XLAPrecision` instead.

    r   r   r   Nc                 ó<   •— t        d«       t        ‰| �	  d¬«       y )NzfThe `TPUPrecision` class is deprecated. Use `lightning_fabric.plugins.precision.XLAPrecision` instead.z32-true©Ú	precisionr   r   s      €r   r   zTPUPrecision.__init__H   s    ø€ ÜØtô	
ô 	‰Ñ 9ÐÕ-r   r,   r(   s   @r   r/   r/   A   s'   ø„ ñð.˜cð .¨Sð .°T÷ .ñ .r   r/   c                   ó0   ‡ — e Zd ZdZdededdfˆ fd„Zˆ xZS )ÚXLABf16Precisionr0   r   r   r   Nc                 ó<   •— t        d«       t        ‰| �	  d¬«       y )NzjThe `XLABf16Precision` class is deprecated. Use `lightning_fabric.plugins.precision.XLAPrecision` instead.z	bf16-truer2   r   r   s      €r   r   zXLABf16Precision.__init__V   s    ø€ ÜØxô	
ô 	‰Ñ ;ÐÕ/r   r,   r(   s   @r   r5   r5   O   s'   ø„ ñð0˜cð 0¨Sð 0°T÷ 0ñ 0r   r5   c                   ó0   ‡ — e Zd ZdZdededdfˆ fd„Zˆ xZS )ÚTPUBf16Precisionr0   r   r   r   Nc                 ó:   •— t        d«       t        ‰| �  |i |¤Ž y )NzjThe `TPUBf16Precision` class is deprecated. Use `lightning_fabric.plugins.precision.XLAPrecision` instead.r   r   s      €r   r   zTPUBf16Precision.__init__d   s"   ø€ ÜØxô	
ô 	‰Ñ˜$Ð) &Ó)r   r,   r(   s   @r   r8   r8   ]   r-   r   r8   c                  ó0  — t        t        j                  dt        «       t        t        j                  dt
        «       t        t        j                  dt        «       t        t        j                  j                  dt        «       t        t        j                  dt        «       t        t        j                  j                  dt        «       t        t        j                  dt        «       t        t        j                  j                  dt        «       y )Nr   r*   r/   r8   r5   )ÚsetattrÚfabricÚ
strategiesr   Úacceleratorsr*   Úpluginsr/   r3   r8   r5   © r   r   Ú_patch_classesrA   k   s£   € ÜŒF×ÑÐ2Ô4EÔFÜŒF×ÑÐ!1´>ÔBÜŒF�N‰N˜N¬LÔ9ÜŒF�N‰N×$Ñ$ n´lÔCÜŒF�N‰NÐ.Ô0@ÔAÜŒF�N‰N×$Ñ$Ð&8Ô:JÔKÜŒF�N‰NÐ.Ô0@ÔAÜŒF�N‰N×$Ñ$Ð&8Ô:JÕKr   )r   N)r
   Útypingr   Úlightning_fabricr<   Úlightning_fabric.acceleratorsr   Ú"lightning_fabric.plugins.precisionr   Úlightning_fabric.strategiesr   Ú&lightning_fabric.strategies.single_xlar   Ú$lightning_fabric.utilities.rank_zeror   r   r   r*   r/   r5   r8   rA   r"   r=   ÚSTRATEGY_REGISTRYr@   r   r   Ú<module>rJ      s“   ðó Ý ã !Ý 8Ý ;Ý 9Ý JÝ FóEôqÐ/ô qô"*�^ô *ô.�<ô .ô0�|ô 0ô*Ð'ô *óLñ Ô Ù Ô à × %Ñ % f×&7Ñ&7×&IÑ&IÕ Jr   