Skip to content

Commit c2acf90

Browse files
authored
Update llama.cpp - ikawrakow#495
1 parent 611e41a commit c2acf90

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9841,7 +9841,7 @@ llm_expert_gating_func_type gating_op,
98419841
}
98429842

98439843
ggml_tensor * par;
9844-
if (lctx.cparams.fused_moe_up_gate) {
9844+
if (lctx.cparams.fused_moe_up_gate && up_exps->type == gate_exps->type) {
98459845
par = ggml_moe_up_gate(ctx, up_exps, gate_exps, cur, selected_experts, type_op == LLM_FFN_SILU ? GGML_UNARY_OP_SILU : GGML_UNARY_OP_GELU);
98469846
} else {
98479847
ggml_tensor * up = llm_build_lora_mm_id(lctx, ctx, up_exps, cur, selected_experts); // [n_ff, n_expert_used, n_tokens]

0 commit comments

Comments
 (0)