Fix __init__.py: remove private imports

2026-05-13 21:43:47 +00:00
parent 1a452ffabd
commit 94233c4dd3
1 changed files with 0 additions and 6 deletions
--- a/src/nvfp4_megamoe_kernel/init.py
+++ b/src/nvfp4_megamoe_kernel/init.py
@@ -7,9 +7,6 @@ from nvfp4_megamoe_kernel.nvfp4_mega_moe import (
    stage_activation,
 )
 from nvfp4_megamoe_kernel.weight_transform import (
-    fold_global_scale,
-    pack_ue4m3_to_uint32,
-    interleave_l1_weights,
    transform_nvfp4_weights_for_tilelang as transform_nvfp4_weights_for_mega_moe,
 )
 from nvfp4_megamoe_kernel.symm_buffer import (
@@ -23,9 +20,6 @@ __all__ = [
    "nvfp4_mega_moe_l2",
    "stage_activation",
    "transform_nvfp4_weights_for_mega_moe",
-    "fold_global_scale",
-    "pack_ue4m3_to_uint32",
-    "interleave_l1_weights",
    "SymmBuffer",
    "get_symm_buffer_for_nvfp4_mega_moe",
 ]