U yh@s(ddlZdgZddZGdddZdS)NLinearBlockSparsePatterncCs |dkr|dkp|dko|dkS)N)row_block_sizecol_block_sizerrT/var/www/html/venv/lib/python3.8/site-packages/torch/ao/nn/sparse/quantized/utils.py%_is_valid_linear_block_sparse_patternsr c@sJeZdZeZdZdZdZdZ d ddZ ddZ ddZ e d d Zd S) rrrcCs8t||sttjtjt_tjt_|t_|t_dSN) r AssertionErrorrrlockacquirerprev_row_block_sizerprev_col_block_size)selfrrrrr __init__s  z!LinearBlockSparsePattern.__init__cCsdSr r)rrrr __enter__!sz"LinearBlockSparsePattern.__enter__cCstjt_tjt_tjdSr )rrrrrr release)rexc_type exc_valueZ backtracerrr __exit__$sz!LinearBlockSparsePattern.__exit__cCs tjtjfSr )rrrrrrr block_size)sz#LinearBlockSparsePattern.block_sizeN)rr)__name__ __module__ __qualname__ threadingRLockr rrrrrrr staticmethodrrrrr rs )r__all__r rrrrr s