1#if !defined(GGML_ROPE_PARAMS)
 2#define GGML_ROPE_PARAMS
 3
 4#include "rte.glsl"
 5
 6struct rope_params {
 7    uint rope_mode;
 8    uint nrows;
 9    uint n_dims;
10    float freq_scale;
11    float freq_base;
12    float ext_factor;
13    float attn_factor;
14    float corr_dims[2];
15    float theta_scale;
16    uint has_ff;
17    int sections[4];
18    uint is_imrope;
19    uint is_back;
20    uint set_rows_stride;
21
22    uint ne00;
23    uint ne01;
24    uint ne02;
25    uint nb01;
26    uint nb02;
27    uint nb03;
28    uint nb11;
29    uint nb12;
30    uint nb13;
31};
32
33#endif // !defined(GGML_ROPE_PARAMS)