Skip to content

Commit b4c0549

Browse files
truecoder34plotnikov.v10
andauthored
ggml-zendnn : fixed naming of matmul function (ggml-org#20964)
* ggml-zendnn: fixed naming of matmul function * ggml-zendnn: fixed naming of mul_mat_id function * ggml-zendnn: fixed print in mul_mat_id --------- Co-authored-by: plotnikov.v10 <plotnikov.v10@wb.ru>
1 parent 0d18aaa commit b4c0549

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

ggml/src/ggml-zendnn/ggml-zendnn.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ static bool ggml_zendnn_matmul(ggml_backend_zendnn_context * ctx, int64_t m, int
8888
return true;
8989
}
9090

91-
static bool ggml_zendnn_sgemm(ggml_backend_zendnn_context * ctx, int64_t m, int64_t n, int64_t k,
91+
static bool ggml_zendnn_gemm(ggml_backend_zendnn_context * ctx, int64_t m, int64_t n, int64_t k,
9292
const void * A, int64_t lda, const void * B, int64_t ldb, void * C,
9393
int64_t ldc, int Atype, int Btype, int Ctype) {
9494

@@ -200,7 +200,7 @@ static void ggml_zendnn_compute_forward_mul_mat(
200200
for (int64_t i12 = 0; i12 < ne12; i12++) {
201201
const void* wdata = (src1->type == vec_dot_type || src0->type == GGML_TYPE_Q8_0) ? src1->data : work_data;
202202
const size_t row_size = ggml_row_size(vec_dot_type, ne10);
203-
if (!ggml_zendnn_sgemm(ctx,
203+
if (!ggml_zendnn_gemm(ctx,
204204
ne01, // m
205205
ne11, // n
206206
ne10, // k
@@ -213,7 +213,7 @@ static void ggml_zendnn_compute_forward_mul_mat(
213213
src0->type,
214214
src0->type == GGML_TYPE_Q8_0 ? GGML_TYPE_F32 : vec_dot_type,
215215
dst->type))
216-
GGML_ABORT("%s: ZenDNN sgemm failed\n", __func__);
216+
GGML_ABORT("%s: ZenDNN gemm failed\n", __func__);
217217
}
218218
}
219219
}
@@ -355,7 +355,7 @@ static void ggml_zendnn_compute_forward_mul_mat_id(
355355
}
356356

357357
// batched gemm for all tokens in this expert
358-
if (!ggml_zendnn_sgemm(ctx,
358+
if (!ggml_zendnn_gemm(ctx,
359359
ne01, // m
360360
cne1, // n
361361
ne10, // k
@@ -368,7 +368,7 @@ static void ggml_zendnn_compute_forward_mul_mat_id(
368368
src0->type,
369369
src0->type == GGML_TYPE_Q8_0 ? GGML_TYPE_F32 : vec_dot_type,
370370
dst->type)) {
371-
GGML_ABORT("%s: ZenDNN sgemm failed\n", __func__);
371+
GGML_ABORT("%s: ZenDNN gemm failed\n", __func__);
372372
}
373373

374374
// scatter output rows to destination

0 commit comments

Comments
 (0)