graph : restore ubatch in build_cb

ggml-ci
2025-07-18 00:27:31 +00:00 · 2025-02-12 16:29:15 +02:00
parent f63aeecce6
commit 6ee86e5e0f
4 changed files with 6 additions and 5 deletions
--- a/src/llama-context.h
+++ b/src/llama-context.h
@ -85,6 +85,7 @@ struct llama_context : public llama_graph_i {
    virtual void build_cb(
             ggml_tensor * cur,
              const char * name,
+      const llama_ubatch & ubatch,
                     int   il);

    // TODO: add encode/decode graphs