o
    TÃi¯  ã                   @   sN   d Z ddlZddlmZ ddlmZ ddlmZ dd„ Z	e
d	kr%e	ƒ  dS dS )
zM
Functionality of swapping optimizer tensors to/from (NVMe) storage devices.
é    Né   )Úaio_basic_multiprocessing)Úaio_handle_multiprocessing)Úget_validated_argsc                  C   s6   t dƒ tƒ } t d¡ | jrtnt}|| | jƒ d S )Nz%Testing deepspeed_aio python frontendÚspawn)Úprintr   ÚmpÚset_start_methodÚhandler   r   Úread)ÚargsÚmultiprocess_function© r   úN/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/nvme/test_ds_aio.pyÚ
ds_io_main   s
   
r   Ú__main__)Ú__doc__Úmultiprocessingr   Úds_aio_basicr   Úds_aio_handler   Úds_aio_argsr   r   Ú__name__r   r   r   r   Ú<module>   s   	
ÿ