ggml : bug fix in ggml_repeat()

2 years ago · 18210579e4
parent 0eea547ab3
commit 18210579e4
1 changed files with 1 additions and 1 deletions
--- a/ggml.c
+++ b/ggml.c
@ -2386,7 +2386,7 @@ struct ggml_tensor * ggml_repeat(
    result->op   = GGML_OP_REPEAT;
    result->grad = is_node ? ggml_dup_tensor(ctx, result) : NULL;
    result->src0 = a;
-    result->src1 = NULL;
+    result->src1 = b;

    return result;
 }