flash_cosine_sim_attention
