summaryrefslogtreecommitdiff
path: root/tests/test-sampling.cpp
diff options
context:
space:
mode:
authorCebtenzzre <cebtenzzre@gmail.com>2023-09-15 15:38:27 -0400
committerGitHub <noreply@github.com>2023-09-15 15:38:27 -0400
commit3aefaab9e59335ebb07d5205dbc8633efd680e58 (patch)
tree1249b50bd748a1bdcc85d010f44feda0d884fef7 /tests/test-sampling.cpp
parent69eb67e28275cd2d57693405f768754a7b2245ad (diff)
check C++ code with -Wmissing-declarations (#3184)
Diffstat (limited to 'tests/test-sampling.cpp')
-rw-r--r--tests/test-sampling.cpp38
1 files changed, 14 insertions, 24 deletions
diff --git a/tests/test-sampling.cpp b/tests/test-sampling.cpp
index 4437c394..019c0d46 100644
--- a/tests/test-sampling.cpp
+++ b/tests/test-sampling.cpp
@@ -12,7 +12,8 @@
#include <vector>
#include <algorithm>
-void dump(const llama_token_data_array * candidates) {
+
+static void dump(const llama_token_data_array * candidates) {
for (size_t i = 0; i < candidates->size; i++) {
printf("%d: %f (%f)\n", candidates->data[i].id, candidates->data[i].p, candidates->data[i].logit);
}
@@ -21,9 +22,7 @@ void dump(const llama_token_data_array * candidates) {
#define DUMP(__candidates) do { printf("%s:%d (%s)\n", __FILE__, __LINE__, __func__); dump((__candidates)); printf("-\n"); } while(0)
-void test_top_k(const std::vector<float> & probs,
- const std::vector<float> & expected_probs,
- int k) {
+static void test_top_k(const std::vector<float> & probs, const std::vector<float> & expected_probs, int k) {
size_t n_vocab = probs.size();
std::vector<llama_token_data> candidates;
candidates.reserve(n_vocab);
@@ -45,10 +44,7 @@ void test_top_k(const std::vector<float> & probs,
}
-void test_top_p(const std::vector<float> & probs,
- const std::vector<float> & expected_probs,
- float p) {
-
+static void test_top_p(const std::vector<float> & probs, const std::vector<float> & expected_probs, float p) {
size_t n_vocab = probs.size();
std::vector<llama_token_data> candidates;
candidates.reserve(n_vocab);
@@ -70,9 +66,7 @@ void test_top_p(const std::vector<float> & probs,
}
-void test_tfs(const std::vector<float> & probs,
- const std::vector<float> & expected_probs,
- float z) {
+static void test_tfs(const std::vector<float> & probs, const std::vector<float> & expected_probs, float z) {
size_t n_vocab = probs.size();
std::vector<llama_token_data> candidates;
candidates.reserve(n_vocab);
@@ -93,9 +87,7 @@ void test_tfs(const std::vector<float> & probs,
}
-void test_typical(const std::vector<float> & probs,
- const std::vector<float> & expected_probs,
- float p) {
+static void test_typical(const std::vector<float> & probs, const std::vector<float> & expected_probs, float p) {
size_t n_vocab = probs.size();
std::vector<llama_token_data> candidates;
candidates.reserve(n_vocab);
@@ -116,11 +108,10 @@ void test_typical(const std::vector<float> & probs,
}
-void test_repetition_penalty(
- const std::vector<float> & probs,
- const std::vector<llama_token> & last_tokens,
- const std::vector<float> & expected_probs,
- float penalty) {
+static void test_repetition_penalty(
+ const std::vector<float> & probs, const std::vector<llama_token> & last_tokens,
+ const std::vector<float> & expected_probs, float penalty
+) {
assert(probs.size() == expected_probs.size());
size_t n_vocab = probs.size();
@@ -145,11 +136,10 @@ void test_repetition_penalty(
}
-void test_frequency_presence_penalty(
- const std::vector<float> & probs,
- const std::vector<llama_token> & last_tokens,
- const std::vector<float> & expected_probs,
- float alpha_frequency, float alpha_presence) {
+static void test_frequency_presence_penalty(
+ const std::vector<float> & probs, const std::vector<llama_token> & last_tokens,
+ const std::vector<float> & expected_probs, float alpha_frequency, float alpha_presence
+) {
assert(probs.size() == expected_probs.size());
size_t n_vocab = probs.size();