
    Vh                         U d dl mZ d dlZg dZ	 d dlmZ daee   e	d<   da
ee   e	d<   ed Znd Zd	ee   fd
Zd	efdZd	ee   fdZy# e$ r dZY ?w xY w)    )OptionalN)versionis_availableget_max_alg_id)_cusparselt__cusparselt_version__MAX_ALG_IDc                  ~    t         7t        j                         a t         dk(  rdayt         dk(  rdayt         dk(  rday)Ni     i     iZ  %   T)r   r   getVersionIntr	        R/home/dcms/DCMS/lib/python3.12/site-packages/torch/backends/cusparselt/__init__.py_initr      sQ      '#.#<#<#> #s* 
 	 &,   &,!r   c                       y)NFr   r   r   r   r   r   &   s    r   returnc                  $    t               syt        S )z Return the version of cuSPARSELtN)r   r   r   r   r   r   r   *   s    7r   c                  6    t         j                  j                  S )z>Return a bool indicating if cuSPARSELt is currently available.)torch_C_has_cusparseltr   r   r   r   r   1   s    88###r   c                  $    t               sy t        S )N)r   r	   r   r   r   r   r   6   s    7r   )typingr   r   __all__torch._Cr   ImportErrorr   int__annotations__r	   r   r   boolr   r   r   r   r   <module>r"      s     $ '+ hsm *"hsm " #  $d $
 O  Ks   A AA