llama : add tensor name for "result_norm" (#9907)

Signed-off-by: Molly Sophia <redacted>

diff --git a/src/llama.cpp b/src/llama.cpp
index 8d44c73c8c95c1dce612e25f350043dac6ecc19a..c51b49c563156c8ed9c4b414beec52da4f381479 100644 (file)
--- a/src/llama.cpp
+++ b/src/llama.cpp
@@ -16095,9 +16095,11 @@ struct llm_build_context {
         cur = ggml_get_rows(ctx0, cur, inp_out_ids);
 
         cur = llm_build_norm(ctx0, cur, hparams, model.output_norm, model.output_norm_b, LLM_NORM, cb, -1);
-        cur = llm_build_lora_mm(lctx, ctx0, model.output, cur);
+        cb(cur, "result_norm", -1);
 
+        cur = llm_build_lora_mm(lctx, ctx0, model.output, cur);
         cb(cur, "result_output", -1);
+
         ggml_build_forward_expand(gf, cur);
 
         return gf;