ggml : check types in release builds too in pool_2d

This commit is contained in:
slaren 2024-01-30 14:49:55 +01:00
parent caf2fc8294
commit bdf3b8ad70

4
ggml.c
View file

@ -12790,8 +12790,8 @@ static void ggml_compute_forward_pool_2d(
const struct ggml_compute_params * params,
const struct ggml_tensor * src,
struct ggml_tensor * dst) {
assert(src->type == GGML_TYPE_F32);
assert(params->ith == 0);
GGML_ASSERT(src->type == GGML_TYPE_F32);
GGML_ASSERT(params->ith == 0);
if (params->type == GGML_TASK_INIT || params->type == GGML_TASK_FINALIZE) {
return;