Multiple updates and refactorings (#231)

2025-11-21 17:49:47 +08:00
parent bb4424aad4
commit 38f8ef73a4
80 changed files with 3767 additions and 2103 deletions
--- a/deep_gemm/utils/layout.py
+++ b/deep_gemm/utils/layout.py
@@ -1,4 +1,4 @@
-from deep_gemm_cpp import (
+from .._C import (
    get_tma_aligned_size,
    get_mk_alignment_for_contiguous_layout,
    get_mn_major_tma_aligned_tensor,