o
    Tix                     @   s*   d dl mZ d dlZdddZdd ZdS )    )versionNc                 C   sT   | s|sJ dt tj}| rt t| |krdS |r(t t||k r(dS dS )Nz2Must provide a min_version or max_version argumentFT)pkg_versionparsetorch__version__str)min_versionmax_versiontorch_version r   I/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/utils/torch.pyrequired_torch_version   s   r   c                 C   s8   t ddr
| |S | | }|jjd d }||S )Ng @)r   r   )r   "register_post_accumulate_grad_hook	expand_asgrad_fnnext_functionsregister_hook)paramhook	param_tmpgrad_accr   r   r   register_grad_hook   s
   



r   )NN)	packagingr   r   r   r   r   r   r   r   r   <module>   s   
