Skip to content

Commit 2871dfa

Browse files
committed
add assert about size
1 parent 447e7b4 commit 2871dfa

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ggml/src/ggml-cpu/ops.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6216,7 +6216,7 @@ static void ggml_compute_forward_conv_2d_impl(const ggml_compute_params * params
62166216

62176217
const int64_t patch_per_thread = (patch_n + params->nth - 1) / params->nth;
62186218
const int64_t patch_start = patch_start_batch + params->ith * patch_per_thread;
6219-
const int64_t patch_end = std::min(patch_start + patch_per_thread,patch_end_batch);
6219+
const int64_t patch_end = std::min(patch_start + patch_per_thread, patch_end_batch);
62206220

62216221
//im2col for a patch
62226222
for (int64_t p = patch_start; p < patch_end; ++p) {
@@ -6258,6 +6258,8 @@ static void ggml_compute_forward_conv_2d_impl(const ggml_compute_params * params
62586258

62596259
float * gemm_output = (float *) ((char *) tmp + patches_per_batch * knl_n * traits->type_size);
62606260

6261+
GGML_ASSERT(gemm_output + patch_n * c_out <= (float*)tmp + params->wsize);
6262+
62616263
// GEMM: patches[patch_n, knl_n] × kernel[knl_n, c_out] = output[patch_n, c_out]
62626264
ggml_call_mul_mat(kernel_type, params, patch_n, c_out, knl_n, tmp, knl_data, gemm_output);
62636265

0 commit comments

Comments
 (0)