o
    TÃi3  ã                   @   s$   d dl mZmZ G dd„ deƒZdS )é    )ÚABCÚabstractmethodc                       s6   e Zd Zd‡ fdd„	Zedd„ ƒZedd„ ƒZ‡  ZS )	Ú	CUDAGraphFc                    s   t ƒ  ¡  || _d S )N)ÚsuperÚ__init__Úenable_cuda_graph)Úselfr   ©Ú	__class__© úg/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/model_implementations/features/cuda_graph.pyr      s   

zCUDAGraph.__init__c                 C   ó   t ‚)z&
        Create CUDA graph(s)
        ©ÚNotImplementedError©r   r   r   r   Ú_create_cuda_graph   ó   zCUDAGraph._create_cuda_graphc                 C   r   )z&
        Replay CUDA graph(s)
        r   r   r   r   r   Ú_graph_replay   r   zCUDAGraph._graph_replay)F)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   Ú__classcell__r   r   r	   r   r   	   s    
r   N)Úabcr   r   r   r   r   r   r   Ú<module>   s   