refactor causal attention masking to support block masks

this change takes the causal masking work from yesterday and generalises it toward block sparse attention, aka FlexAttention, by making the causal or custom mask merely one of the supported sparsity patterns.