
    9j                     \   U d dl Z d dlZd dlZd dlZd dlmZmZmZ ej                  j                         Z
 G d de j                        Z G d de j                        Zi dej                  dej                  d	ej                  d
ej                  dej                  dej                  dej                  dej                  dej                  dej                  dej                  dej                   dej                   dej                   dej                   dej                   dej                   ej                   ej                   ej                   ej                   ej                   ej                   dZej                  ej$                  ej                   ej&                  iZ ej*                  d      dedz  fd       Z ej*                  d      dedz  fd       Z ed       Zd Z ed       Zeed <   e
s#ej                  j                         r ed!      yy)"    N)
IS_WINDOWSLazyValTEST_XPUc                       e Zd ZdZdZy)XPUCodenamePVCBMGN)__name__
__module____qualname__r   r	        b/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/testing/_internal/common_xpu.pyr   r      s    
C
Cr   r   c                       e Zd ZdZdZdZy)XPUArchr         N)r
   r   r   UnknownXeXe2r   r   r   r   r      s    G	
B
Cr   r   i  i  i  i  i  i  i  i  i  ii  in  i  i  i  i  i  i  )i  i  i   i!  i"  i#  r   returnc                  n    t         j                  j                         d   } t        j	                  |       S )N	device_id)torchxpuget_device_capability_DEVICE_ID_TO_CODENAMEget)r   s    r   get_xpu_codenamer   :   s*    		//1+>I!%%i00r   c                  ^    t               } t        j                  | t        j                        S N)r   _CODENAME_TO_ARCHr   r   r   )codenames    r   get_xpu_archr$   @   s!    !H  7??;;r   c                  x    t         j                  j                         xr t               t        j
                  k\  S r!   )r   r   is_availabler$   r   r   r   r   r   <lambda>r'   G   s#    EII""$F7;;)F r   c                  .    t         rt         xr t        S y)NF)r   r   Xe2_Or_Laterr   r   r   *evaluate_platform_supports_flash_attentionr*   K   s    ~.,.r   c                      t               S r!   )r*   r   r   r   r'   r'   R   s
    68 r   %PLATFORM_SUPPORTS_FLASH_ATTENTION_XPUz'XPU should not be initialized on import)enum	functoolsr   	torch.xpu$torch.testing._internal.common_utilsr   r   r   r   is_initialized!XPU_ALREADY_INITIALIZED_ON_IMPORTEnumr   IntEnumr   r   r	   r   r   r   r"   	lru_cacher   r$   r)   r*   r,   bool__annotations__AssertionErrorr   r   r   <module>r9      sd       N N %*II$<$<$> !$)) 
dll 
KOO
KOO KOO KOO	
 KOO KOO KOO KOO KOO KOO KOO KOO KOO KOO KOO  KOO!" KOO#$ OOOOOOOOOOOO/ 8 OOWZZOOW[[  Q1+, 1 1
 Q<gn < <
 F
 /68/ %t 
 )yy!FGG " )r   