Searched defs:seqlenq_ngroups_swapped (Results 1 – 1 of 1) sorted by relevance
48 void set_params_fprop(Flash_fwd_params ¶ms, in set_params_fprop()408 …const int seqlenq_ngroups_swapped = seqlen_q == 1 && num_heads > num_heads_k && window_size_left <… in mha_fwd() local620 …const int seqlenq_ngroups_swapped = max_seqlen_q == 1 && num_heads > num_heads_k && window_size_le… in mha_varlen_fwd() local1330 …const int seqlenq_ngroups_swapped = seqlen_q == 1 && num_heads > num_heads_k && window_size_left <… in mha_fwd_kvcache() local