opencl: do not treat NULL-mask flash attention as causal (#25771)

This commit is contained in:
kumaal
2026-07-24 08:12:01 -07:00
committed by GitHub
parent 77095ee0cb
commit 56a83860dd
+5 -1
View File
@@ -14758,7 +14758,11 @@ static void ggml_cl_flash_attn(ggml_backend_t backend, const ggml_tensor * q, co
// Flash-Decoding K-split decision. Resolved here, before the prefill
// prepass, because KV-pad and blk prepass are pure overhead when FD fires.
const int is_causal = (mask == NULL && n_q > 1 && n_q == n_kv);
// Do not infer causality from tensor shapes: a NULL mask means full
// (bidirectional) attention, e.g. ViT encoders, where n_q == n_kv as well.
// Causal attention in llama.cpp always comes with an explicit KQ mask.
// Inferring is_causal here corrupted mmproj output on OpenCL (see #23800).
const int is_causal = 0;
const int fd_max_n_q = (d_head_q <= FD_MAX_DK_MULTI) ? FD_MAX_N_Q_MULTI : 1;
cl_kernel fd_k_split = NULL;
bool use_fd_mq = false;