o
    Zh                     @   s   U d dl mZ d dlZg dZzd dlmZ W n ey"   dZY nw daee e	d< da
ee e	d< edur<dd Znd	d Zd
ee fddZd
efddZd
ee fddZdS )    )OptionalN)versionis_availableget_max_alg_id)_cusparselt__cusparselt_version__MAX_ALG_IDc                   C   s@   t d u rt a t dkrdadS t dkrdadS t dkrdadS )Ni     i     iZ  %   T)r   r   ZgetVersionIntr    r   r   Q/var/www/auris/lib/python3.10/site-packages/torch/backends/cusparselt/__init__.py_init   s   r   c                   C   s   dS )NFr   r   r   r   r   r   &   s   returnc                   C   s   t  sdS tS )z Return the version of cuSPARSELtN)r   r   r   r   r   r   r   *   s   r   c                   C   s   t jjS )z>Return a bool indicating if cuSPARSELt is currently available.)torchZ_CZ_has_cusparseltr   r   r   r   r   1   s   r   c                   C   s   t  sd S tS )N)r   r   r   r   r   r   r   6   s   r   )typingr   r   __all__Ztorch._Cr   ImportErrorr   int__annotations__r   r   r   boolr   r   r   r   r   r   <module>   s    
