xla

[ragged-paged-attn] Apply kv mask to filter out NaNs

#9219

Merged