From d9fe17bdad1a84cdfc8e9f0acc2e7ef40079d24c Mon Sep 17 00:00:00 2001 From: Gaurav Garg Date: Wed, 15 Jul 2026 16:15:07 +0530 Subject: [PATCH] Fix tests on Vulkan --- tests/test-backend-sampler.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test-backend-sampler.cpp b/tests/test-backend-sampler.cpp index a76cf192ba..5a1475f8ac 100644 --- a/tests/test-backend-sampler.cpp +++ b/tests/test-backend-sampler.cpp @@ -732,7 +732,7 @@ static void test_backend_multi_sequence_sampling(const test_params & params) { } static void test_backend_dist_sampling(const test_params & params) { - const int seq_id = 189; + const int seq_id = 0; const int32_t seed = 88; struct llama_sampler_chain_params backend_chain_params = llama_sampler_chain_default_params(); @@ -1801,7 +1801,7 @@ static void test_backend_multi_output_sampling_chain(const test_params & params) const int32_t seed = 88; const llama_vocab * vocab = llama_model_get_vocab(params.model.get()); const int32_t n_vocab = llama_vocab_n_tokens(vocab); - const uint32_t k = std::min(2048, n_vocab); + const uint32_t k = std::min(512, n_vocab); const llama_logit_bias bias = { llama_vocab_bos(vocab), -0.1f };