Skip to content

Commit e32d081

Browse files
authored
ggml : always check bounds on get_rows operations (#9354)
1 parent df270ef commit e32d081

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ggml/src/ggml.c

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

13724-
assert(i01 >= 0 && i01 < ne01);
13724+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1372513725

1372613726
dequantize_row_q(
1372713727
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13762,7 +13762,7 @@ static void ggml_compute_forward_get_rows_f16(
1376213762
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1376313763
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1376413764

13765-
assert(i01 >= 0 && i01 < ne01);
13765+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1376613766

1376713767
ggml_fp16_to_fp32_row(
1376813768
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13803,7 +13803,7 @@ static void ggml_compute_forward_get_rows_bf16(
1380313803
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1380413804
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1380513805

13806-
assert(i01 >= 0 && i01 < ne01);
13806+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1380713807

1380813808
ggml_bf16_to_fp32_row(
1380913809
(const void *) ((char *) src0->data + i01*nb01 + i11*nb02 + i12*nb03),
@@ -13844,7 +13844,7 @@ static void ggml_compute_forward_get_rows_f32(
1384413844
const int64_t i10 = (i - i12*ne11*ne10 - i11*ne10);
1384513845
const int64_t i01 = *(int32_t *) ((char *) src1->data + i10*nb10 + i11*nb11 + i12*nb12);
1384613846

13847-
assert(i01 >= 0 && i01 < ne01);
13847+
GGML_ASSERT(i01 >= 0 && i01 < ne01);
1384813848

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

0 commit comments

Comments
 (0)