From: Xiao-Yong Jin <redacted>
Date: Sun, 26 Nov 2023 08:30:02 +0000 (-0600)
Subject: metal : fix yarn (#4220)
X-Git-Tag: upstream/0.0.4488~2921
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=22da05536ff4ad963080773bef1fb839fdab95d3;p=pkg%2Fggml%2Fsources%2Fllama.cpp

metal : fix yarn (#4220)

get the correct n_orig_ctx in metal
---

diff --git a/ggml-metal.m b/ggml-metal.m
index a9fdd390..d52a1c3c 100644
--- a/ggml-metal.m
+++ b/ggml-metal.m
@@ -1433,7 +1433,8 @@ void ggml_metal_graph_compute(
                             const int n_past     = ((int32_t *) dst->op_params)[0];
                             const int n_dims     = ((int32_t *) dst->op_params)[1];
                             const int mode       = ((int32_t *) dst->op_params)[2];
-                            const int n_orig_ctx = ((int32_t *) dst->op_params)[3];
+                            // skip 3, n_ctx, used in GLM RoPE, unimplemented in metal
+                            const int n_orig_ctx = ((int32_t *) dst->op_params)[4];
 
                             float freq_base, freq_scale, ext_factor, attn_factor, beta_fast, beta_slow;
                             memcpy(&freq_base,   (int32_t *) dst->op_params +  5, sizeof(float));