use I64 for indices

Author: rgerganov

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

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