Home
last modified time | relevance | path

Searched defs:AccumLambdaIteratorSm70 (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.h96 struct AccumLambdaIteratorSm70 { struct
102 using Policy = typename T::Policy;
103 using InstructionShape = typename T::InstructionShape;
104 using OpDelta = typename T::OpDelta;
105 using Shape = typename T::Shape;
106 using Element = accum_t;
108 static int const kElementsPerPartial = 4;
109 using EleShapePerPatial = typename cutlass::platform::conditional<
113 static int const kElementsPerMma = 8;
114 static int const kAccumulatorPatials = 2;
[all …]