Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dkernel_traits.h18 struct Flash_kernel_traits { struct
21 using Element = elem_type;
22 static constexpr bool Has_cp_async = true;
28 using ElementAccum = float;
52 typename Base=Flash_kernel_traits<kHeadDim_, kBlockM_, kBlockN_, kNWarps_, elem_type> > argument