
    *`i	                     "    d dl Z d dlZdgZd ZdS )    Nget_permutations_24c           	         g }t          d          D ]g }dz  }|dz  }dD ]\}ddz  z  ddz  z  dz   ddz  dz   z  ddz  dz   z  dz   fD ]/}|                    d|z  |dz  z   d|dz  z  z   d|z  z              0]t          d          D ]#|                    fd	|D                        $t          j        |          }| dk    rt          j        g d
          }n?| dk    rt          j        g d          }n"t          d                    |                     |                    dt          |          f          d d |f         	                                }t          j        |          }g }	t          d          D ]#|	                    fddD                        $g }
t          d          D ]#|
                    fddD                        $||	|
fS )N          )r      r            c                      g | ]
}|d z  z   S )r    ).0pjs     |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/compressed_tensors/utils/permutations_24.py
<listcomp>z'get_permutations_24.<locals>.<listcomp>/   s!    777Aa!a%i777    )r   r   r      r            )r   r   r   r   znum_bits must be 4 or 8, got {}c                      g | ]
}d z  |z   S r   r   r   r   is     r   r   z'get_permutations_24.<locals>.<listcomp>=   s!    GGG1q519GGGr   )r   r   r   r   r   r   r   r   c                      g | ]
}d z  |z   S r   r   r   s     r   r   z'get_permutations_24.<locals>.<listcomp>@   s!    !N!N!N!a%!)!N!N!Nr   )r   r   r   r   r   r   r   r   )rangeappendextendnumpyarray
ValueErrorformatreshapelenraveltorch
from_numpy)num_bits	perm_listperm1colcol_oblockrowperm
interleave
scale_permscale_perm_singler   r   s              @@r   r   r       so   I2YY 9 91fq 	Q 	QEQUQUaQUQYQUQY!#	 Q Q R#X3a37mCa%iOPPPPQ q 	9 	9A77777778888	9;y!!D1}}[!9!9!9::

	Q[..

:AA(KKLLL<<S__-..qqq*}=CCEEDD!!DJ1XX I IGGGG.FGGGHHHH1XX P P  !N!N!N!N5M!N!N!NOOOO...r   )r!   r(   __all__r   r   r   r   <module>r6      s<       !
!!/ !/ !/ !/ !/r   