o
    Ti                     @   s   d dl Z d dlmZ zd dlmZ W n ey1   zd dlmZ W n ey.   dd ZY nw Y nw dd Zdd	 Z	d
d ZdS )    Nrequired_torch_version)is_compilingc                   C   s   dS )NF r   r   r   N/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/runtime/compiler.py<lambda>   s    r   c                   C   s
   t ddS )Ng @)min_versionr   r   r   r   r   is_compile_supported   s   
r	   c                 C   s   t  r	tj| S | S N)r	   torchcompilerdisable)funcr   r   r   r      s   r   c                   C   s   t  S r
   )torch_is_compilingr   r   r   r   r      s   r   )
r   deepspeed.utils.torchr   torch.compilerr   r   ImportErrortorch._dynamo.external_utilsr	   r   r   r   r   r   <module>   s   