From f2cd962fe28b6dbb78b97ffe961fbae34c1468bb Mon Sep 17 00:00:00 2001 From: Radoslav Gerganov Date: Fri, 20 Jun 2025 11:37:43 +0300 Subject: [PATCH] use I64 for indices --- ggml/src/ggml-cpu/ops.cpp | 2 +- ggml/src/ggml.c | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ggml/src/ggml-cpu/ops.cpp b/ggml/src/ggml-cpu/ops.cpp index 2bbaa5a5d..c618fd5b9 100644 --- a/ggml/src/ggml-cpu/ops.cpp +++ b/ggml/src/ggml-cpu/ops.cpp @@ -4501,7 +4501,7 @@ static void ggml_compute_forward_set_rows_f32( const int64_t i12 = i/(ne11*ne10); const int64_t i11 = (i - i12*ne11*ne10)/ne10; const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10); - const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12); + const int64_t i01 = *(int64_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12); GGML_ASSERT(i01 >= 0 && i01 < ne1); diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c index 8e99c5523..13405a7d3 100644 --- a/ggml/src/ggml.c +++ b/ggml/src/ggml.c @@ -3408,7 +3408,7 @@ struct ggml_tensor * ggml_set_rows( GGML_ASSERT(c->ne[3] == 1); GGML_ASSERT(a->type == GGML_TYPE_F16); GGML_ASSERT(b->type == GGML_TYPE_F32); - GGML_ASSERT(c->type == GGML_TYPE_I32); + GGML_ASSERT(c->type == GGML_TYPE_I64); struct ggml_tensor * result = ggml_view_tensor(ctx, a);