CUTLASS: reduce_split_k.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• include

• cutlass

• reduction

• kernel

Classes | Namespaces

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.](reduce split k_8h_source.html)

|

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 | | |

---

Generated by 1.8.11