use I64 for indices

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

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