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