fix: module name in probe_moe_fused_breakdown.sh

2026-08-15 12:43:55 +00:00
parent b0af7d54ff
commit e47b66e268
1 changed files with 1 additions and 1 deletions
--- a/qwen3_6_scripts/probe_moe_fused_breakdown.sh
+++ b/qwen3_6_scripts/probe_moe_fused_breakdown.sh
@@ -6,7 +6,7 @@ import torch
 import time
 import importlib.util

-spec = importlib.util.spec_from_file_location('m',
+spec = importlib.util.spec_from_file_location('corex_batched_gemm',
    'qwen3_6_scripts/prebuilt/corex-3.2.3-ivcore10/corex_batched_gemm.so')
 mod = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(mod)