From a465dd1d75667471e82c349c0de36219242ebb16 Mon Sep 17 00:00:00 2001 From: Claude Date: Sat, 15 Aug 2026 05:22:31 +0000 Subject: [PATCH] fix: use F.silu in test script for old corex torch --- ex_engine/csrc/build_test_moe_tcu.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ex_engine/csrc/build_test_moe_tcu.sh b/ex_engine/csrc/build_test_moe_tcu.sh index 4163599d..2784fa89 100755 --- a/ex_engine/csrc/build_test_moe_tcu.sh +++ b/ex_engine/csrc/build_test_moe_tcu.sh @@ -66,7 +66,7 @@ for k in range(top_k): eid = expert_ids[k].item() w = expert_weights[k].item() gate_up = F.linear(hidden, w13[eid]) - gate = torch.silu(gate_up[:, :I]) + gate = F.silu(gate_up[:, :I]) up = gate_up[:, I:] act = gate * up expert_out = F.linear(act, w2[eid]) @@ -131,7 +131,7 @@ for _ in range(3): eid = expert_ids[k].item() w = expert_weights[k].item() gate_up = F.linear(hidden, w13[eid]) - gate = torch.silu(gate_up[:, :I]) + gate = F.silu(gate_up[:, :I]) up = gate_up[:, I:] act = gate * up out_py += w * F.linear(act, w2[eid]) @@ -144,7 +144,7 @@ for _ in range(100): eid = expert_ids[k].item() w = expert_weights[k].item() gate_up = F.linear(hidden, w13[eid]) - gate = torch.silu(gate_up[:, :I]) + gate = F.silu(gate_up[:, :I]) up = gate_up[:, I:] act = gate * up out_py += w * F.linear(act, w2[eid])