Skip to content

Commit 9a55680

Browse files
slarenggerganov
authored andcommitted
ggml : always check bounds on get_rows operations (llama/9354)
1 parent 778181a commit 9a55680

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/ggml.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13709,7 +13709,7 @@ static void ggml_compute_forward_get_rows_q(
1370913709
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1371013710
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1371113711

13712-
assert(i01 >= 0 && i01 < ne01);
13712+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1371313713

1371413714
dequantize_row_q(
1371513715
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13750,7 +13750,7 @@ static void ggml_compute_forward_get_rows_f16(
1375013750
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1375113751
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1375213752

13753-
assert(i01 >= 0 && i01 < ne01);
13753+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1375413754

1375513755
ggml_fp16_to_fp32_row(
1375613756
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13791,7 +13791,7 @@ static void ggml_compute_forward_get_rows_bf16(
1379113791
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1379213792
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1379313793

13794-
assert(i01 >= 0 && i01 < ne01);
13794+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1379513795

1379613796
ggml_bf16_to_fp32_row(
1379713797
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13832,7 +13832,7 @@ static void ggml_compute_forward_get_rows_f32(
1383213832
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1383313833
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1383413834

13835-
assert(i01 >= 0 && i01 < ne01);
13835+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1383613836

1383713837
ggml_vec_cpy_f32(nc,
1383813838
(float *) ((char *) dst->data + i10*nb1 + i11*nb2 + i12*nb3),

0 commit comments

Comments
 (0)