flash_cosine_sim_attention
flash_cosine_sim_attention_cuda_0_2_2
