
    nVh                     ,    d dl mZ dZd Zdee   fdZy)    )Listi   c                     | | dz
  z  dk(  S )N   r    )xs    F/home/dcms/DCMS/lib/python3.12/site-packages/triton/language/_utils.pyis_power_of_twor	      s    QKA    shapec           	         d}t        |       D ]O  \  }}t        |t              st        d| dt	        |       d      t        |      st        d| d      ||z  }Q |t        kD  rt        d| dt         d      |S )	Nr   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 ())	enumerate
isinstanceint	TypeErrortyper	   
ValueErrorTRITON_MAX_TENSOR_NUMEL)r   numelids       r   validate_block_shaper   
   s    E%  1!S!nQC/`aefgah`iijkllq!~aS0EFGG
 &&75')PQhPiijkllLr
   N)typingr   r   r	   r   r   r   r
   r   <module>r      s#    ! S	 r
   