yukuai e82c4139da Revert "Fixed the bug in get_swizzle_mode function related to elem_size setting. (#115)" ... This reverts commit ac428e25e0. This PR causes wgrad to hang during testing. Revert it until we resolve the issue		2025-06-23 17:13:36 +08:00
..
__init__.py	Weight gradient kernels for dense and MoE models (#95)	2025-05-14 14:47:58 +08:00
gemm.py	Revert "Fixed the bug in get_swizzle_mode function related to elem_size setting. (#115)"	2025-06-23 17:13:36 +08:00
m_grouped_gemm.py	Fix JIT tests	2025-05-16 14:39:58 +08:00
runtime.py	Unify kwargs usages	2025-05-15 16:53:52 +08:00
utils.py	Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs	2025-03-13 22:15:16 +08:00
wgrad_gemm.py	Fix JIT tests	2025-05-16 14:39:58 +08:00