]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
ggml : fix ggml_clamp (#263)
authorLoganDark <redacted>
Fri, 16 Jun 2023 19:17:30 +0000 (12:17 -0700)
committerGitHub <redacted>
Fri, 16 Jun 2023 19:17:30 +0000 (22:17 +0300)
This unconditionally failed before

src/ggml.c

index a3f116cc4518278b9eba3ea6a7825311cbf5b515..25f46caedfbf8eff50e13208fca525bea5abadc2 100644 (file)
@@ -6453,7 +6453,7 @@ struct ggml_tensor * ggml_clamp(
 
     ggml_scratch_save(ctx);
 
-    struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_F32, 3);
+    struct ggml_tensor * b = ggml_new_tensor_1d(ctx, GGML_TYPE_F32, 2);
 
     ((float *) b->data)[0] = min;
     ((float *) b->data)[1] = max;
@@ -11263,7 +11263,7 @@ static void ggml_compute_forward_clamp_f32(
         const struct ggml_tensor * src1,
         struct ggml_tensor * dst) {
     assert(params->ith == 0);
-    assert(src1->type == GGML_TYPE_I32);
+    assert(src1->type == GGML_TYPE_F32);
     assert(ggml_nelements(src1) == 2);
 
     if (params->type == GGML_TASK_INIT || params->type == GGML_TASK_FINALIZE) {