Searched defs:is_sm90 (Results 1 – 1 of 1) sorted by relevance
366 bool is_sm90 = dprops->major == 9 && dprops->minor == 0; in mha_fwd() local566 bool is_sm90 = dprops->major == 9 && dprops->minor == 0; in mha_varlen_fwd() local834 bool is_sm90 = dprops->major == 9 && dprops->minor == 0; in mha_bwd() local1052 bool is_sm90 = dprops->major == 9 && dprops->minor == 0; in mha_varlen_bwd() local1276 bool is_sm90 = dprops->major == 9 && dprops->minor == 0; in mha_fwd_kvcache() local