Äntligen, riktig blixtuppmärksamhet. Med den nya indexeringen är det helt automatiskt och generiskt, se TestFuse.test_flash_attention. ENDRANGE är STORE+LOAD, och du avslutar ett intervall om det är underordnat ett UOp-matchningsfel på den axeln.