Searched defs:Flash_kernel_traits (Results 1 – 1 of 1) sorted by relevance
18 struct Flash_kernel_traits { struct21 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