“The correctness tests for FlashKDA verify an exact match against a PyTorch reference implementation.”