diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c index a5796214f25..15c4d01c6a2 100644 --- a/ggml/src/ggml.c +++ b/ggml/src/ggml.c @@ -3687,6 +3687,7 @@ struct ggml_tensor * ggml_set_rows( result->op = GGML_OP_SET_ROWS; result->src[0] = b; result->src[1] = c; + result->src[2] = a; // make sure a gets a backend buffer return result; }