
    3j                     t   % S SK r S SKrS SKrS SKrS SKJrJrJr  \R                  R                  5       r
 " S S\ R                  5      r " S S\ R                  5      r0 S\R                  _S\R                  _S	\R                  _S
\R                  _S\R                  _S\R                  _S\R                  _S\R                  _S\R                  _S\R                  _S\R                  _S\R                   _S\R                   _S\R                   _S\R                   _S\R                   _S\R                   _\R                   \R                   \R                   \R                   \R                   \R                   S.Er\R                  \R$                  \R                   \R&                  0r\R*                  " S5      S\S-  4S j5       r\R*                  " S5      S\S-  4S j5       r\" S 5      rS r\" S 5      r\\S '   \
(       d(  \R                  R                  5       (       a  \" S!5      egg)"    N)
IS_WINDOWSLazyValTEST_XPUc                       \ rS rSrSrSrSrg)XPUCodename   PVCBMG N)__name__
__module____qualname____firstlineno__r	   r
   __static_attributes__r       \/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/testing/_internal/common_xpu.pyr   r      s    
C
Cr   r   c                        \ rS rSrSrSrSrSrg)XPUArch   r         r   N)r   r   r   r   UnknownXeXe2r   r   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    [         R                  R                  5       S   n [        R	                  U 5      $ )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                  ^    [        5       n [        R                  U [        R                  5      $ N)r#   _CODENAME_TO_ARCHr"   r   r   )codenames    r   get_xpu_archr(   @   s!    !H  7??;;r   c                      [         R                  R                  5       =(       a    [        5       [        R
                  :  $ r%   )r   r   is_availabler(   r   r   r   r   r   <lambda>r+   G   s$    EII""$F7;;)FFr   c                  J    [         (       a  [        (       + =(       a    [        $ g)NF)r   r   Xe2_Or_Laterr   r   r   *evaluate_platform_supports_flash_attentionr.   K   s    x~.,.r   c                      [        5       $ r%   )r.   r   r   r   r+   r+   R   s    68r   %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.   r0   bool__annotations__AssertionErrorr   r   r   <module>r=      sc       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   