Home
last modified time | relevance | path

Searched defs:access_m_idx (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
Dwarp_iterator_from_smem.h176 int access_m_idx = ldsm_vec_num % kTilesPerInstruction; in WarpIteratorFromSmem() local
260 for (int access_m_idx = 0; access_m_idx < in load() local