a
    þd‹  ã                   @   s  d dl mZmZmZmZ d dlZd dlmZ d dlmZ e	dœdd„Z
edœdd	„Zeeeed
œdd„Zeeeed
œdd„Ze e
ƒ ¡e d¡kr d dlmZ nd dlmZ eddd ƒrÒesÐee e	dœdd„Zn>erüdee ee	 eedf dœdd„Znee e	dœdd„ZdS )é    )ÚTYPE_CHECKINGÚListÚOptionalÚTupleN)ÚTensor)Úversion)Úreturnc                   C   s   t j d¡d S )z<Parse the `torch.__version__` variable and removes +cu*/cpu.ú+r   )ÚtorchÚ__version__Úsplit© r   r   ú]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/utils/_compat.pyÚtorch_version	   s    r   c                 C   s$   t  tƒ ¡}|t  | › d|› ¡kS ©NÚ.©r   Úparser   )ÚmajorÚminorÚ_versionr   r   r   Útorch_version_geq   s    r   )r   r   Úpatchr   c                 C   s*   t  tƒ ¡}|t  | › d|› d|› ¡k S r   r   ©r   r   r   r   r   r   r   Útorch_version_lt   s    r   c                 C   s*   t  tƒ ¡}|t  | › d|› d|› ¡kS r   r   r   r   r   r   Útorch_version_ge   s    r   z1.7.1)Úqré   é
   ©ÚtensorsÚindexingc                 C   s   t j| |dS )N)r!   ©r
   Zmeshgridr   r   r   r   Útorch_meshgrid+   s    r#   .)r    r!   r   c                 C   s
   t  | ¡S ©Nr"   r   r   r   r   r#   2   s    c                 C   s
   t  | ¡S r$   r"   r   r   r   r   r#   7   s    )N)Útypingr   r   r   r   r
   r   Ú	packagingr   Ústrr   Úboolr   Úintr   r   r   Ztorch.linalgr   Z	linalg_qrr#   r   r   r   r   Ú<module>   s    &