| 182 | } |
| 183 | |
| 184 | static void test_top_n_sigma(const std::vector<float> & probs, const std::vector<float> & probs_expected, int n) { |
| 185 | sampler_tester tester(probs, probs_expected); |
| 186 | |
| 187 | DUMP(&tester.cur_p); |
| 188 | tester.apply(llama_sampler_init_top_n_sigma(n)); |
| 189 | tester.apply(llama_sampler_init_dist (0)); |
| 190 | DUMP(&tester.cur_p); |
| 191 | |
| 192 | tester.check(); |
| 193 | } |
| 194 | |
| 195 | static void test_sampler_queue(const size_t n_vocab, const std::string & samplers_sequence, const int top_k, const float top_p, const float min_p |
| 196 | ) { |
no test coverage detected