Наконец, настоящая быстрая внимательность. С новым индексированием это полностью автоматизировано и универсально, см. TestFuse.test_flash_attention. ENDRANGE — это STORE+LOAD, и вы завершаете диапазон, если дочерние элементы UOp не совпадают по этой оси.
92,7K