From: goerch <redacted>
Date: Tue, 11 Jul 2023 17:39:54 +0000 (+0200)
Subject: ggml : fix ggml_set_xxx (#354)
X-Git-Tag: upstream/0.0.1642~1337
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=61fe6df55753498f08503466491270e75e7c9f59;p=pkg%2Fggml%2Fsources%2Fggml

ggml : fix ggml_set_xxx (#354)

`ggml_set_i32`/`ggml_set_f32` are in line with `ggml_set_i32_1d`/`ggml_set_f32_1d` then.
---

diff --git a/src/ggml.c b/src/ggml.c
index 487ad948..6419cc8b 100644
--- a/src/ggml.c
+++ b/src/ggml.c
@@ -4723,7 +4723,7 @@ struct ggml_tensor * ggml_set_i32 (struct ggml_tensor * tensor, int32_t value) {
             {
                 assert(tensor->nb[0] == sizeof(ggml_fp16_t));
                 for (int i = 0; i < n; i++) {
-                    ggml_vec_set_f16(nc, (ggml_fp16_t *)(data + i*n1), value);
+                    ggml_vec_set_f16(nc, (ggml_fp16_t *)(data + i*n1), GGML_FP32_TO_FP16(value));
                 }
             } break;
         case GGML_TYPE_F32:
@@ -4775,7 +4775,7 @@ struct ggml_tensor * ggml_set_f32(struct ggml_tensor * tensor, float value) {
             {
                 assert(tensor->nb[0] == sizeof(ggml_fp16_t));
                 for (int i = 0; i < n; i++) {
-                    ggml_vec_set_f16(nc, (ggml_fp16_t *)(data + i*n1), value);
+                    ggml_vec_set_f16(nc, (ggml_fp16_t *)(data + i*n1), GGML_FP32_TO_FP16(value));
                 }
             } break;
         case GGML_TYPE_F32: