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