fast path

2025-09-15 18:57:13 -04:00
parent 479b6144df
commit 125e7b5fe6
8 changed files with 79 additions and 16 deletions
--- a/scripts/bench_moe.py
+++ b/scripts/bench_moe.py
@@ -156,7 +156,6 @@ def main():
    )

    # HF Triton (stub compute for now)
-    os.environ.setdefault("AXOLOTL_MOE_BACKEND", "hf_triton")
    t_hf = forward_hf_triton
    y = t_hf(x, gate, experts, args.top_k)
    if y is not None: