From da1e744efd1b2d945cf509abd8a10db3d6f9562d Mon Sep 17 00:00:00 2001
From: Sergio Lopez <slp@redhat.com>
Date: Mon, 10 Feb 2025 11:54:51 +0100
Subject: [PATCH] tests: test softmax with ne00 == 2048

It was found that softmax vulkan shaders may fail on some contexts when
ne00 > 1024, so let's add a test for it.

Signed-off-by: Sergio Lopez <slp@redhat.com>
---
 tests/test-backend-ops.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp
index 1bfd41254..5c562aa90 100644
--- a/tests/test-backend-ops.cpp
+++ b/tests/test-backend-ops.cpp
@@ -4166,7 +4166,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
         for (float max_bias : {0.0f, 8.0f}) {
             if (!mask && max_bias > 0.0f) continue;
             for (float scale : {1.0f, 0.1f}) {
-                for (int64_t ne0 : {16, 1024}) {
+                for (int64_t ne0 : {16, 1024, 2048}) {
                     for (int64_t ne1 : {16, 1024}) {
                         if (mask) {
                             for (ggml_type m_prec : {GGML_TYPE_F32, GGML_TYPE_F16}) {