Write the reference before debugging the shader

The Flash Attention 2 path was easier to finish after the algorithm existed in plain JavaScript...