[Bugfix] Disable FlexAttention direct block mask building for encoder-only models (vllm-project#27344)

Signed-off-by: Isotr0py <[email protected]>
Signed-off-by: Alberto Perdomo <[email protected]>

Author: Isotr0py

vllm/v1/attention/backends/flex_attention.py

@@ -658,7 +658,10 @@ def build(
             total_cache_tokens=total_cache_tokens,
             decode_offset=offset_tensor,
             num_blocks_per_seq=num_blocks_per_seq,
-            direct_build=self.direct_build,
+            # FIXME(Isotr0py): direct build has issue to build bidirectional
+            # attention block mask for encoder-only models, disable it temporarily.
+            # see: https://github.com/vllm-project/vllm/pull/27329#issuecomment-3431484053
+            direct_build=(self.direct_build and common_attn_metadata.causal),
             q_block_size=self.q_block_size,
             kv_block_size=self.kv_block_size,
         )