flash_cosine_sim_attention
flash_cosine_sim_attention_cuda
