feat: add sm90 megamoe phase1 interfaces

2026-06-17 23:54:49 +08:00
parent 062cb160cf
commit 1dafd0f173
12 changed files with 1520 additions and 130 deletions
@@ -85,6 +85,7 @@ from .mega import (
    SymmBuffer,
    get_symm_buffer_for_mega_moe,
    transform_weights_for_mega_moe,
+    fp8_mega_moe,
    fp8_fp4_mega_moe,
 )