+
    ғLj                    r  a 0 t $ ^ RIHt ^ RIHt ^ RIHtHtHtH	t	H
t
Ht ]'       dC   ^RIHt ]	]],          ]]R3,          ]P                  ]P                   3,          t]]R3,          tRtR R ltR	 R
 ltR tRPR R lltR R ltR tR R lt/ RRbRRbRRbRRbRRbRRbRRbRRbR R!bR"R!bR#RbR$R%bR&R%bR'R(bR)R(bR*R+bR,R-bR.R-R/R0R1R0R2R3R4R5R6R7R8R7R9R:R;R<R=R>R?R@RARBRCRC/Ct]! ]P;                  4       4       F  t]]]&   K
  	  RD tRE t / RQ U u/ uF	  p RFV  2V bK  	  up CRQ U u/ uF	  p RGV  2V bK  	  up CRR U u/ uF	  p RHV  2V bK  	  up CRS Uu/ uF	  pRIV 2^bK  	  upCR+^RC^ /Ct!RJ]"RK&   ]PG                  4        F  w  t$t]!],          ]!]$&   K  	  RL RM lt%RN t&RO t'R# u up i u up i u up i u upi )T    )annotationsreduce)AnyCallableTYPE_CHECKINGUnionListDictcore.i   c               $    V ^8  d   QhRRRRRR/# )   iterableIterableTypepathObjPathreturnr    )formats   "9/app/.local/lib/python3.14/site-packages/triton/_utils.py__annotate__r      s!     9 9 9G 9 9    c                    \        R  W4      # )c                    W,          # Nr   )aidxs   &&r   <lambda>#get_iterable_path.<locals>.<lambda>   s    r   r   )r   r   s   &&r   get_iterable_pathr!      s    '88r   c               $    V ^8  d   QhRRRRRR/# )r   r   r   r   tuple[int, ...]valr   r   )r   s   "r   r   r      s!     ! ! !O !# !r   c                    ^RI Hp \        V4      ^ 8w  g   Q h\        V4      ^8X  d   T M\        WRR 4      p\	        WCP
                  4      '       g   Q hVP                  VR,          V4       R# )   r   N)languager   lenr!   
isinstancetuple_setitem)r   r   r$   r   prevs   &&&  r   set_iterable_pathr.      sY    t9>>4yA~8+<XCRy+QDdJJ''''MM$r(C r   c                f    ^RI Hp \        V \        \        VP                  VP
                  34      # )r&   r   )r(   r   r*   listr+   
tuple_type)xr   s   & r   is_iterabler3      s#    a$tzz4??CDDr   Nc               $    V ^8  d   QhRRRRRR/# )r   valuer   fnzCallable[[ObjPath, Any], None]r   Noner   )r   s   "r   r   r      s#      3 $B SW r   c                    Vf   Rp\        V 4      '       d)   \        V 4       F  w  r4\        WA. VOVN5R7       K  	  R # V! W 4       R # )N)_pathr   )r3   	enumerateapply_with_path)r5   r6   r9   r   items   &&&  r   r;   r;      sD    }5"5)ICDMeMSM: * 	5r   c               $    V ^8  d   QhRRRRRR/# )r   r   zUnion[IterableType, Any]predzCallable[[ObjPath, Any], bool]r   zlist[ObjPath]r   )r   s   "r   r   r   *   s$      4 <Z _l r   c                f   aaa / oR  VVV3R lloS! RV 4       \        SP                  4       4      # )c                    V ^8  d   QhRRRR/# )r   r   r#   currentr   r   )r   s   "r   r   #find_paths_if.<locals>.__annotate__.   s      O c r   c                   < \        V4      '       d%   \        V4       F  w  r#S! . V OVN5V4       K  	  R # S! W4      '       d   R SV &   R # R # r   )r3   r:   )r   rA   r   r<   _implr>   rets   &&  r   rD   find_paths_if.<locals>._impl.   sJ    w&w/	llclD) 0$  CI !r   r   )r0   keys)r   r>   rD   rE   s   &f@@r   find_paths_ifrH   *   s/    !C  
"h
r   c                &    W ^,
          ,          ^ 8H  # )r&   r   )r2   s   &r   is_power_of_tworJ   :   s    QKAr   c                   V ^8  d   QhRR/# )r   shapez	List[int]r   )r   s   "r   r   r   >   s      	 r   c           	     &   ^p\        V 4       F^  w  r#\        V\        4      '       g   \        RV R\	        V4       R24      h\        V4      '       g   \        RV R24      hW,          pK`  	  V\        8  d   \        RV R\         R24      hV# )r&   zShape element z1 must have type `constexpr[int]`, got `constexpr[]z must be a power of 2znumel (z') exceeds triton maximum tensor numel ())r:   r*   int	TypeErrortyperJ   
ValueErrorTRITON_MAX_TENSOR_NUMEL)rL   numelids   &   r   validate_block_shaperX   >   s    E% !S!!nQC/`aefgah`iijkllq!!~aS0EFGG
 ! &&75')PQhPiijkllLr   boolu1int1uint1i1
float8e4nvfp8e4nvfloat8e5fp8e5float8e4b15fp8e4b15float8_e4m3fn
float8e4b8fp8e4b8float8_e4m3fnuzfloat8_e5m2float8e5b16fp8e5b16float8_e5m2fnuzhalffp16float16bfloat16bf16floatfp32float32doublefp64float64int8i8int16i16rP   i32int32int64i64uint8u8uint16u16uint32u32uint64u64voidc                ^    \        V 4      P                  R 4      R,          p\        V,          # ).r'   )strsplittype_canonicalisation_dict)dtype	dtype_strs   & r   canonicalize_dtyper   u   s&    E
  %b)I%i00r   c                4    V'       d   R MR \        V 4       2# )z*k*)r   )r   is_consts   &&r   canonicalize_ptr_dtyper   z   s    dC();E)B(CDDr   urV   fpfp8zDict[str, int]BITWIDTH_DICTc                    V ^8  d   QhRRRR/# )r   r   r   r   rP   r   )r   s   "r   r   r      s        #  #  r   c                    \         V ,          # r   )r   )r   s   &r   get_primitive_bitwidthr      s    r   c                z    \        V \        4      ;'       d%    \        V \        4      ;'       d    \	        V R 4      # _fields)r*   rR   
issubclassr+   hasattr)r$   s   &r   is_namedtupler      s-    c4 WWZU%;WWY@WWr   c                f    \        V R 4      '       d   \        V 4      ! V!  # \        V 4      ! V4      # r   )r   rR   )argcontentss   &&r   _tuple_creater      s-     $+3	#:#:49hSS	(@SSr   r   )r&             @   )r   r   r   )e4nve4b15e4b8e5e5b16)(__conditional_annotations__
__future__r   	functoolsr   typingr   r   r   r	   r
   r   r(   r   r0   r+   r1   r   rP   r   rT   r!   r.   r3   r;   rH   rJ   rX   r   valuesvr   r   r   __annotations__itemskr   r   r   )nsuffixr   s   00@r   <module>r      s   " "  B BcE#s(OTZZPQLCHoG! 9!E
 #
D# D# T	#
 	$# )# # :# Y# )# y# 7# :# z#  F!#" v##$ %#& V'#( vfv
DU	5UUTeee
FE# J 
(//1	2A$%q! 
31
E!"$"1 
1#wz"$! #$"1 
1#wz"$! 1 
A3x{! >?=6 
VH~q=?! B
A!~  ',,.DAq$Q'M! / XT1$$?s   F%'F*=F/F4