Finalmente, atenção real em flash. Com a nova indexação, é completamente automática e genérica, veja TestFuse.test_flash_attention. ENDRANGE é STORE+LOAD, e você termina um intervalo se os filhos de um UOp não coincidirem nesse eixo.
92,75K