Searched defs:max_smem_per_block (Results 1 – 2 of 2) sorted by relevance
141 int max_smem_per_block; in run_mha_bwd_hdim32() local165 int max_smem_per_block; in run_mha_bwd_hdim64() local210 int max_smem_per_block; in run_mha_bwd_hdim96() local236 int max_smem_per_block; in run_mha_bwd_hdim128() local270 int max_smem_per_block; in run_mha_bwd_hdim160() local290 int max_smem_per_block; in run_mha_bwd_hdim192() local318 int max_smem_per_block; in run_mha_bwd_hdim256() local
323 int max_smem_per_block; in run_mha_fwd_hdim224() local352 int max_smem_per_sm, max_smem_per_block; in run_mha_fwd_hdim256() local