
    bi9                     8    d dl Z d dlmZ d Zedk(  r e        yy)    N)is_xpu_availablec                     d} d}t         j                  j                         r!t         j                  j                         }d} n*t	               r t         j
                  j                         }d} t        d| d|  d       y )NGPUr   XPUzSuccessfully ran on  s)torchcudais_availabledevice_countr   xpuprint)accelerator_typenum_acceleratorss     a/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/accelerate/test_utils/scripts/test_cli.pymainr      sr    zz  ::224 		 99113 	 !1 2!4D3EQ
GH    __main__)r	   accelerate.utilsr   r   __name__ r   r   <module>r      s'     -	I zF r   