bash_unit allows you to write unit tests (functions starting with test), run them and, in case of failure, displays the stack trace with source file and line number indications to locate the problem.

from bitblas.tl.utils import make_mma_swizzle_layout as make_swizzle_layout from bitblas.tl.mma_macro_generator import ( B_shared_shape = (block_N, block_K // num ...