o
    TÃik  ã                   @   sH   d dl mZ d dlT d dlmZ eeeee	e
eeeeeeiZdd„ ZdS )é    )Úget_scalar_param)Ú*)Úget_acceleratorc                 C   s’   t |  ¡ v rG| t  d urG| t  }tt|ttƒtt|ttƒtt|ttƒt	t|t	t
ƒtt|ttƒtt|ttƒi}|t rEtƒ  ¡ dksEJ dƒ‚|S tS )NÚcudaz1GDS currently only supported for CUDA accelerator)ÚAIOÚkeysÚAIO_BLOCK_SIZEr   ÚAIO_BLOCK_SIZE_DEFAULTÚAIO_QUEUE_DEPTHÚAIO_QUEUE_DEPTH_DEFAULTÚAIO_INTRA_OP_PARALLELISMÚ AIO_INTRA_OP_PARALLELISM_DEFAULTÚAIO_SINGLE_SUBMITÚAIO_SINGLE_SUBMIT_DEFAULTÚAIO_OVERLAP_EVENTSÚAIO_OVERLAP_EVENTS_DEFAULTÚAIO_USE_GDSÚAIO_USE_GDS_DEFAULTr   Údevice_nameÚAIO_DEFAULT_DICT)Ú
param_dictÚaio_dictÚ
aio_config© r   ú\/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/runtime/swap_tensor/aio_config.pyÚget_aio_config   s&   





ôr   N)Údeepspeed.runtime.config_utilsr   Ú'deepspeed.runtime.swap_tensor.constantsÚdeepspeed.acceleratorr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Ú<module>   s   ú
