Commit a464960ed for llama.cpp
commit a464960ed34564913a94afab17bf79fde727023e
Author: Aaron Teo <aaron.teo1@ibm.com>
Date: Tue Nov 3 23:51:07 2026 +0800
ggml-zdnn: fix fattn test failure (#31583)
diff --git a/ggml/src/ggml-zdnn/fattn.cpp b/ggml/src/ggml-zdnn/fattn.cpp
index 7c7801cbb..e877b7a75 100644
--- a/ggml/src/ggml-zdnn/fattn.cpp
+++ b/ggml/src/ggml-zdnn/fattn.cpp
@@ -20,8 +20,8 @@ bool ggml_zdnn_supports_flash_attn_ext(const ggml_backend_zdnn_device_context *
memcpy(&max_bias, (const float *) op->op_params + 1, sizeof(float));
memcpy(&logit_softcap, (const float *) op->op_params + 2, sizeof(float));
- // sinks, alibi, softcap and sparse k/v sets are not implemented
- if (op->src[4] != nullptr || max_bias != 0.0f || logit_softcap != 0.0f || ggml_get_op_params_i32(op, 4) != 0) {
+ // sinks, alibi, softcap, sparse k/v sets and k/v row lists are not implemented
+ if (op->src[4] != nullptr || op->src[5] != nullptr || max_bias != 0.0f || logit_softcap != 0.0f || ggml_get_op_params_i32(op, 4) != 0) {
return false;
}