
    bi                        d dl Z d dlmZ d dlZ	 d dlZd dlmZ dZ ej                  g dej                   eed      rej                  j                         rdnd      Z ej                  g d	ej                   eed      rej                  j                         rdnd      Zeed
Zd Z e       Zy# e	$ r dZY w xY w)    N)versionTF)      g    6Gg    fg    TFٿg   I4ҿg   ০ǿg    O        g   __?g   `\?g   ?g   @g?g    4?g   ` ?g   `v"?      ?xpucpu)dtypedevice)r   g_Lu?gK=U?r   gioT?g      ?g-!lV?g      ?r   g_LugK=Ur   gioTտg      g-!lVſg      п)nf4fp4c                      t        j                  dddd      } | j                  j                         syt	        j
                  | j                  j                  d      d   j                         d         S )z4
    Returns the installed version of Gaudi SW.
    z#pip list | grep habana-torch-pluginT)shelltextcapture_outputN
r   )
subprocessrunstdoutstripr   parsesplit)outputs    V/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/bitsandbytes/backends/utils.pyget_gaudi_sw_versionr   B   sd     ^^-	F == ==,,T215;;=bABB    )r   	packagingr   torchtritontriton.languagelanguagetltriton_availableImportErrortensorfloat32hasattrr   is_available_NF4_QUANT_TABLE_FP4_QUANT_TABLECODEr   GAUDI_SW_VER r   r   <module>r.      s       
  5<<$ --ue!7!7!9 	- 0  5<<$ --ue!7!7!9 	- 0  (89C" $%Q  s   B8 8CC