Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dkernel_traits.h32 using MMA_Atom_Arch = std::conditional_t< member
38 using MMA_Atom_Arch = MMA_Atom<SM75_16x8x8_F32F16F16F32_TN>; member