From 2880f43b7f6bec91fd61567a6c15736602591e50 Mon Sep 17 00:00:00 2001
From: Alex Renda <alex@renda.org>
Date: Tue, 20 Jun 2023 20:49:43 -0400
Subject: [PATCH] add test for correct top-p behavior

---
 tests/test-sampling.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test-sampling.cpp b/tests/test-sampling.cpp
index 5d693f7b5..64f9455d7 100644
--- a/tests/test-sampling.cpp
+++ b/tests/test-sampling.cpp
@@ -181,6 +181,7 @@ int main(void) {
 
     test_top_p({0.1f, 0.2f, 0.3f, 0.4f}, {0.4f}, 0);
     test_top_p({0.1f, 0.2f, 0.3f, 0.4f}, {0.4f, 0.3f}, 0.7f);
+    test_top_p({0.1f, 0.2f, 0.3f, 0.4f}, {0.4f, 0.3f, 0.2f}, 0.8f);
     test_top_p({0.1f, 0.2f, 0.3f, 0.4f}, {0.4f, 0.3f, 0.2f, 0.1f}, 1);
 
     test_tfs({0.1f, 0.15f, 0.2f, 0.25f, 0.3f}, {0.3f}, 0.25f);