Sebastian Kimberk

Does anyone have a solution for this problem? The previous answer didn't make much sense to me. Thanks.

@yuan-luo I'd check their repo: https://github.com/AlphaGPU/leetgpu-challenges. Looks like they have PyTorch reference solutions (e.g. https://github.com/AlphaGPU/leetgpu-challenges/blob/main/challenges/hard/12_multi_head_attention/challenge.py) although nothing in Triton as far as I can tell.

+1 that the current notation for Puzzle 9 is confusing/potentially incorrect, and also that `B1` is missing As @VachanVY mentioned, it looks like it's been fixed/improved in `Triton-Puzzles-Lite`: https://github.com/SiriusNEO/Triton-Puzzles-Lite/blob/main/puzzles.md#puzzle-9-simple-flashattention $$z_{i}...

Had the same issue when using the Colab notebook, the issue was solved by running @txfs19260817's command (thanks!) It's kindof weird, since theoretically the notebook is trying to install the...

@kunal-mansukhani Looks like Github messed up the url, correct one is https://leetgpu.com/ And for anyone else looking, it looks like there (as of now) 54 challenges which can be solved...

Awesome, thanks so much! Yeah I just started on Puzzle 9 and was struggling to figure out what was going on. Looks like there's been a PR out to fix...