Ë
    P/æjQ  ã                   ó,   — d dl Z dgZd„ Z G d„ d«      Zy)é    NÚLinearBlockSparsePatternc                 ó6   — | dk(  xr |dk(  xs | dk(  xr |dk(  S )Né   é   é   © )Úrow_block_sizeÚcol_block_sizes     úe/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/ao/nn/sparse/quantized/utils.pyÚ%_is_valid_linear_block_sparse_patternr      s0   € Ø˜aÑÒ7 N°aÑ$7ò Ø˜!ÑÒ3 °!Ñ 3ðó    c                   ób   — e Zd Z ej                  «       ZdZdZdZdZ	dd„Z
d„ Zd„ Zed„ «       Zy)	r   r   r   c                 óì   — t        ||«      sJ ‚t        j                  j                  «        t        j                  t        _        t        j                  t        _        |t        _        |t        _        y ©N)r   r   ÚrlockÚacquirer	   Úprev_row_block_sizer
   Úprev_col_block_size)Úselfr	   r
   s      r   Ú__init__z!LinearBlockSparsePattern.__init__   s]   € Ü4°^À^ÔTÐTÐTÜ ×&Ñ&×.Ñ.Ô0ä$×3Ñ3ô 	!Ô4ô %×3Ñ3ô 	!Ô4ð 3AÔ Ô/Ø2@Ô Õ/r   c                  ó   — y r   r   )r   s    r   Ú	__enter__z"LinearBlockSparsePattern.__enter__'   s   € Ør   c                 ó¤   — t         j                  t         _        t         j                  t         _        t         j
                  j                  «        y r   )r   r   r	   r   r
   r   Úrelease)r   Úexc_typeÚ	exc_valueÚ	backtraces       r   Ú__exit__z!LinearBlockSparsePattern.__exit__*   s:   € ä$×8Ñ8ô 	!Ô/ô %×8Ñ8ô 	!Ô/ô 	!×&Ñ&×.Ñ.Õ0r   c                  óB   — t         j                  t         j                  fS r   )r   r	   r
   r   r   r   Ú
block_sizez#LinearBlockSparsePattern.block_size3   s    € ô %×3Ñ3Ü$×3Ñ3ð
ð 	
r   N)r   r   )Ú__name__Ú
__module__Ú__qualname__Ú	threadingÚRLockr   r	   r
   r   r   r   r   r   Ústaticmethodr    r   r   r   r   r      sJ   „ ØˆI�O‰OÓ€EØ€NØ€NØÐØÐó
Aòò1ð ñ
ó ñ
r   )r$   Ú__all__r   r   r   r   r   ú<module>r(      s!   ðã ð &Ð
&€ò÷$
ò $
r   