| 73 | } |
| 74 | |
| 75 | static void test_temp_ext(const std::vector<float> & probs, const std::vector<float> & probs_expected, float temp, float delta, float exponent) { |
| 76 | sampler_tester tester(probs, probs_expected); |
| 77 | |
| 78 | DUMP(&tester.cur_p); |
| 79 | tester.apply(llama_sampler_init_temp_ext(temp, delta, exponent)); |
| 80 | tester.apply(llama_sampler_init_dist (0)); |
| 81 | DUMP(&tester.cur_p); |
| 82 | |
| 83 | tester.check(); |
| 84 | } |
| 85 | |
| 86 | static void test_top_k(const std::vector<float> & probs, const std::vector<float> & probs_expected, int k) { |
| 87 | sampler_tester tester(probs, probs_expected); |
no test coverage detected