Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
H A Dmma_accum_lambda_iterator.h209 struct AccumLambdaIteratorSimt { struct
210 using Policy = typename T::Policy;
211 using Iterations = typename T::Iterations;
212 using Element = typename T::Element;
213 using Delta = typename T::Delta;
214 using Shape = typename T::Shape;
221 CUTLASS_DEVICE static bool reduceSameRow(int lane_id, DT& myValue, F fn) { in reduceSameRow()
231 CUTLASS_DEVICE static void iterateRows( in iterateRows()
263 static cutlass::MatrixCoord CUTLASS_DEVICE get_lane_offset( in get_lane_offset()