Co-authored-by: Judd <foldl@boxvest.com>
This commit is contained in:
Judd 2024-07-03 20:40:16 +08:00 committed by GitHub
parent fadde67135
commit f8d6a23804
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5312,7 +5312,7 @@ void ggml_mul_mat_set_prec(
as -> [cols, rows, n_expert] as -> [cols, rows, n_expert]
ids -> [n_experts_used, n_tokens] (i32) ids -> [n_experts_used, n_tokens] (i32)
b -> [cols, n_expert_used, n_tokens] b -> [cols, n_expert_used, n_tokens]
c -> [cols, n_expert_used, n_tokens] c -> [rows, n_expert_used, n_tokens]
in b, n_experts_used can be broadcasted to match the n_expert_used of ids in b, n_experts_used can be broadcasted to match the n_expert_used of ids