Explorar o código

ggml : fix dependencies for ggml_set_rows (#16318)

Georgi Gerganov hai 3 meses
pai
achega
a4a0aa5ea2
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      ggml/src/ggml.c

+ 1 - 0
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; // note: order is weird due to legacy reasons (https://github.com/ggml-org/llama.cpp/pull/16063#discussion_r2385795931)
 
     return result;
 }