diff options
author | Jed Fox <git@jedfox.com> | 2023-05-06 17:01:47 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-06 17:01:47 -0400 |
commit | 3924088512d9e12e90ed6dbf28a6c5712481d33e (patch) | |
tree | 9dacd76924e57e792ab834b47d77b875d1c8ae4f /tests | |
parent | 173d0e6419e8f8f3c1f4f13201b777f4c60629f3 (diff) |
Remove default arguments from sampling functions (#1343)
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test-sampling.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test-sampling.cpp b/tests/test-sampling.cpp index 8ce59af..9174c1e 100644 --- a/tests/test-sampling.cpp +++ b/tests/test-sampling.cpp @@ -32,7 +32,7 @@ void test_top_k(const std::vector<float> & probs, llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; llama_sample_softmax(nullptr, &candidates_p); DUMP(&candidates_p); - llama_sample_top_k(nullptr, &candidates_p, k); + llama_sample_top_k(nullptr, &candidates_p, k, 1); DUMP(&candidates_p); assert(candidates_p.size == expected_probs.size()); @@ -57,7 +57,7 @@ void test_top_p(const std::vector<float> & probs, llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; llama_sample_softmax(nullptr, &candidates_p); DUMP(&candidates_p); - llama_sample_top_p(nullptr, &candidates_p, p); + llama_sample_top_p(nullptr, &candidates_p, p, 1); DUMP(&candidates_p); assert(candidates_p.size == expected_probs.size()); @@ -80,7 +80,7 @@ void test_tfs(const std::vector<float> & probs, llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; DUMP(&candidates_p); - llama_sample_tail_free(nullptr, &candidates_p, z); + llama_sample_tail_free(nullptr, &candidates_p, z, 1); DUMP(&candidates_p); assert(candidates_p.size == expected_probs.size()); @@ -103,7 +103,7 @@ void test_typical(const std::vector<float> & probs, llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; DUMP(&candidates_p); - llama_sample_typical(nullptr, &candidates_p, p); + llama_sample_typical(nullptr, &candidates_p, p, 1); DUMP(&candidates_p); assert(candidates_p.size == expected_probs.size()); |