@@ -138,7 +138,7 @@ int main(int /*argc*/, const char ** /*argv*/) {
138138 struct ggml_tensor * x;
139139
140140 // rope f32
141- for (int m = 0 ; m < 6 ; ++m) {
141+ for (int m = 0 ; m < 5 ; ++m) {
142142 const int ndims = 4 ;
143143
144144 const int64_t n_rot = 128 ;
@@ -153,7 +153,7 @@ int main(int /*argc*/, const char ** /*argv*/) {
153153 x = get_random_tensor_f32 (ctx0, ndims, ne, -1 .0f , 1 .0f );
154154 int mode = -1 ;
155155
156- if (m < 3 ) {
156+ if (m < 2 ) {
157157 struct ggml_tensor * p0 = ggml_new_tensor_1d (ctx0, GGML_TYPE_I32, ne[2 ]);
158158 struct ggml_tensor * p1 = ggml_new_tensor_1d (ctx0, GGML_TYPE_I32, ne[2 ]);
159159 struct ggml_tensor * p2 = ggml_new_tensor_1d (ctx0, GGML_TYPE_I32, ne[2 ]);
@@ -163,9 +163,8 @@ int main(int /*argc*/, const char ** /*argv*/) {
163163 ((int32_t *) p1->data )[i] = n_past_2 - n_past_0;
164164 ((int32_t *) p2->data )[i] = n_past_2 + i;
165165 }
166- // test mode 0, 2, 4 (standard, GPT-NeoX, GLM)
167- // note: GLM is not implemented, it will default to standard
168- mode = m == 0 ? 0 : m == 1 ? 2 : 4 ;
166+ // test mode 0, 2 (standard, GPT-NeoX)
167+ mode = m == 0 ? GGML_ROPE_TYPE_NORMAL : GGML_ROPE_TYPE_NEOX;
169168
170169 // 100, 101, 102, ..., 172
171170 r0 = ggml_rope (ctx0, x, p0, n_rot, mode);
@@ -181,7 +180,8 @@ int main(int /*argc*/, const char ** /*argv*/) {
181180 struct ggml_tensor * p2 = ggml_new_tensor_1d (ctx0, GGML_TYPE_I32, ne[2 ] * 4 );
182181
183182 int sections[4 ] = {16 , 24 , 24 , 0 };
184- mode = (m == 3 ) ? GGML_ROPE_TYPE_MROPE : (m == 4 ) ? GGML_ROPE_TYPE_VISION : GGML_ROPE_TYPE_IMROPE;
183+
184+ mode = (m == 2 ) ? GGML_ROPE_TYPE_MROPE : (m == 3 ) ? GGML_ROPE_TYPE_VISION : GGML_ROPE_TYPE_IMROPE;
185185
186186 for (int i = 0 ; i < ne[2 ]; ++i) {
187187 for (int j = 0 ; j < 4 ; ++j) {
0 commit comments