Apply suggestions

RWKV · Jun 13, 2023 · 023eda1 · 023eda1
1 parent 0b5ab44
commit 023eda1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_context_cloning.c b/tests/test_context_cloning.c
@@ -42,7 +42,7 @@ int main() {
 
 	rwkv_eval(ctx, prompt[0], NULL, state, logits);
 
-	for (int i = 1; i < prompt[i] != 0; i++) {
+	for (int i = 1; prompt[i] != 0; i++) {
 		rwkv_eval(ctx, prompt[i], state, state, logits);
 	}
 

diff --git a/tests/test_tiny_rwkv.c b/tests/test_tiny_rwkv.c
@@ -33,7 +33,7 @@ void test_model(const char * model_path, const float * expected_logits, const fl
     ASSERT(rwkv_gpu_offload_layers(model, N_GPU_LAYERS), "Unexpected error %d", rwkv_get_last_error(model));
 #endif
 
-    size_t n_vocab = rwkv_get_logits_len(model);
+    const size_t n_vocab = rwkv_get_logits_len(model);
 
     ASSERT(n_vocab == N_VOCAB, "Unexpected n_vocab in the model");