flash_cosine_sim_attention
tests
