diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2024-05-26 18:35:23 +0300 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2024-05-29 14:29:33 +0300 |
commit | 72de268bec49f67e2883880f573c55cea32de736 (patch) | |
tree | c79389a7cdd6684513240c2bd5047c54f7c61b0e /ggml.h | |
parent | 0e8d8bfd6caf1d0a8cbdf9d3d5c06fbbb9dfced8 (diff) |
ggml : restore ggml_rope_xpos_inplace (ggml/0)
ggml-ci
Diffstat (limited to 'ggml.h')
-rw-r--r-- | ggml.h | 8 |
1 files changed, 8 insertions, 0 deletions
@@ -1548,6 +1548,14 @@ extern "C" { float beta_slow), "use ggml_rope_ext_inplace instead"); + struct ggml_tensor * ggml_rope_xpos_inplace( + struct ggml_context * ctx, + struct ggml_tensor * a, + struct ggml_tensor * b, + int n_dims, + float base, + bool down); + // compute correction dims for YaRN RoPE scaling GGML_CALL void ggml_rope_yarn_corr_dims( int n_dims, int n_orig_ctx, float freq_base, float beta_fast, float beta_slow, float dims[2]); |