ソースを参照

ggml : check ggml_add src1 type (ggml/708)

Co-authored-by: Judd <foldl@boxvest.com>
Judd 2 年 前
コミット
e976423005
1 ファイル変更6 行追加1 行削除
  1. 6 1
      ggml.c

+ 6 - 1
ggml.c

@@ -7498,7 +7498,12 @@ static void ggml_compute_forward_add(
     switch (src0->type) {
     switch (src0->type) {
         case GGML_TYPE_F32:
         case GGML_TYPE_F32:
             {
             {
-                ggml_compute_forward_add_f32(params, src0, src1, dst);
+                if (src1->type == GGML_TYPE_F32) {
+                    ggml_compute_forward_add_f32(params, src0, src1, dst);
+                }
+                else {
+                    GGML_ASSERT(false);
+                }
             } break;
             } break;
         case GGML_TYPE_F16:
         case GGML_TYPE_F16:
             {
             {