CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-11-04 09:32:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

ggml : ggml_set_rows update comment + better index name

Browse Source
This commit is contained in:
Georgi Gerganov
2025-06-22 18:45:07 +03:00
parent e89709721b
commit e73690a69d
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ggml/include/ggml.h
View File

@@ -1383,6 +1383,8 @@ extern "C" {
// b TS [n_embd, n_rows, ne02, ne03] | ne02 == ne2, ne03 == ne3 // b TS [n_embd, n_rows, ne02, ne03] | ne02 == ne2, ne03 == ne3
// c I64 [n_rows, ne11, ne12, 1] | c[i] in [0, ne1) // c I64 [n_rows, ne11, ne12, 1] | c[i] in [0, ne1)
// //
// undefined behavior if destination rows overlap
//
// broadcast: // broadcast:
// ne2 % ne11 == 0 // ne2 % ne11 == 0
// ne3 % ne12 == 0 // ne3 % ne12 == 0

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

@@ -4558,13 +4558,13 @@ static void ggml_compute_forward_set_rows_f32(
const int64_t i11 = i02%ne11; const int64_t i11 = i02%ne11;
const int64_t i10 = i; const int64_t i10 = i;
const int64_t i01 = *(int64_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12); const int64_t i1 = *(int64_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
GGML_ASSERT(i01 >= 0 && i01 < ne1); GGML_ASSERT(i1 >= 0 && i1 < ne1);
from_float( from_float(
(const float *) ((char *) src0->data + i*nb01 + i02*nb02 + i03*nb03), (const float *) ((char *) src0->data + i*nb01 + i02*nb02 + i03*nb03),
((char *) dst->data + i01*nb1 + i02*nb2 + i03*nb3), nc); ((char *) dst->data + i1*nb1 + i02*nb2 + i03*nb3), nc);
} }
} }
} }