diff --git a/src/nvfp4_megamoe_kernel/__init__.py b/src/nvfp4_megamoe_kernel/__init__.py index fbaa059e..c592bdb5 100644 --- 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", ]