metal : unify mul_mv_id kernels (#6556)

This commit is contained in:
slaren 2024-04-12 18:13:20 +02:00 committed by GitHub
parent 4cc120c744
commit fbbc030ba9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 208 additions and 1122 deletions

1
ggml.c
View file

@ -11012,7 +11012,6 @@ static void ggml_compute_forward_mul_mat_id(
}
// initialize matrix_row_counts
GGML_ASSERT(wdata == wdata_src1_end);
memset(matrix_row_counts, 0, n_as*sizeof(int64_t));
// group rows by src0 matrix