[DRAFT] Use mask instead of cond for attention conditional logic

[DRAFT] Use mask instead of cond for attention conditional logic #3187