reduce_split_k.h File Reference

Kernel performing a reduction over densely packed tensors in global memory. More...

#include "cutlass/cutlass.h"
#include "cutlass/tensor_ref.h"
#include "cutlass/numeric_types.h"
#include "cutlass/array.h"
#include "cutlass/functional.h"
#include "cutlass/matrix_shape.h"
#include "cutlass/numeric_conversion.h"
#include "cutlass/layout/matrix.h"

Include dependency graph for reduce_split_k.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	cutlass::reduction::kernel::ReduceSplitK< Shape_, OutputOp_, ReductionOp_, PartitionsPerStage >
struct	cutlass::reduction::kernel::ReduceSplitK< Shape_, OutputOp_, ReductionOp_, PartitionsPerStage >::Params
	Params structure. More...
struct	cutlass::reduction::kernel::ReduceSplitK< Shape_, OutputOp_, ReductionOp_, PartitionsPerStage >::SharedStorage

Namespaces
	cutlass
	cutlass::reduction
	cutlass::reduction::kernel