Skip to content

Commit da1e744

Browse files
committed
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 <[email protected]>
1 parent bae9a58 commit da1e744

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test-backend-ops.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4166,7 +4166,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
41664166
for (float max_bias : {0.0f, 8.0f}) {
41674167
if (!mask && max_bias > 0.0f) continue;
41684168
for (float scale : {1.0f, 0.1f}) {
4169-
for (int64_t ne0 : {16, 1024}) {
4169+
for (int64_t ne0 : {16, 1024, 2048}) {
41704170
for (int64_t ne1 : {16, 1024}) {
41714171
if (mask) {
41724172
for (ggml_type m_prec : {GGML_TYPE_F32, GGML_TYPE_F16}) {

0 commit comments

Comments
 (0)