Triton Single-Query Attention Kernel

Unlock premium problems

This is a Premium problem. Upgrade to unlock the full editor, test cases, and solution.