§
    &`ƒi-  ã                   óN   — d dl mZ d dlmZ d dlZd dlmZ  G d„ de¦  «        ZdS )é    ©Úcontextmanager)ÚListN)ÚTorchDeviceManagerc                   ób   — e Zd ZdZd e¦   «         fd„Zdeej                 fd„Z	defd„Z
d„ ZdS )ÚCPUTorchDeviceManagerzCPU device managerÚreturnc                 ó   — dS )NT© ©Úselfs    úx/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/air/_internal/device_manager/cpu.pyÚis_availablez"CPUTorchDeviceManager.is_available   s   € Øˆtó    c                 ó,   — t          j        d¦  «        gS )z?Gets the correct torch device list configured for this process.Úcpu)ÚtorchÚdevicer   s    r   Úget_devicesz!CPUTorchDeviceManager.get_devices   s   € å”˜UÑ#Ô#Ð$Ð$r   c                 ó   — dS )z3Validate if the device type support create a streamFr   r   s    r   Úsupports_streamz%CPUTorchDeviceManager.supports_stream   s   € àˆur   c                 ó6   — t           d„ ¦   «         } |¦   «         S )z&Return empty context mananger for CPU.c               3   ó   K  — d V — d S )Nr   r   r   r   Údefault_context_managerzICPUTorchDeviceManager.get_stream_context.<locals>.default_context_manager   s   è è € àˆEˆEˆEˆEˆEr   r   )r   Ústreamr   s      r   Úget_stream_contextz(CPUTorchDeviceManager.get_stream_context   s.   € õ 
ð	ð 	ñ 
Œð	ð 'Ð&Ñ(Ô(Ð(r   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úboolr   r   r   r   r   r   r   r   r   r   r   r   	   s‡   € € € € € ØÐð˜d˜d™fœfð ð ð ð ð%˜T %¤,Ô/ð %ð %ð %ð %ð ð ð ð ð ð)ð )ð )ð )ð )r   r   )Ú
contextlibr   Útypingr   r   Ú5ray.air._internal.device_manager.torch_device_managerr   r   r   r   r   ú<module>r%      sy   ðØ %Ð %Ð %Ð %Ð %Ð %Ø Ð Ð Ð Ð Ð à €€€à TÐ TÐ TÐ TÐ TÐ Tð)ð )ð )ð )ð )Ð.ñ )ô )ð )ð )ð )r   