tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-08-06 09:10:11 -04:00
Code Issues Packages Projects Releases Wiki Activity

use I64 for indices

Browse Source
This commit is contained in:
Radoslav Gerganov
2025-06-20 11:37:43 +03:00
committed by Georgi Gerganov
parent c1a581a10b
commit f2cd962fe2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ggml/src/ggml-cpu/ops.cpp
View File

@@ -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);

2
ggml/src/ggml.c
View File

@@ -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);