CUTLASS: cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count > Struct Template Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• Class List

• Class Index

• Class Hierarchy

• Class Members

• cutlass

• reduction

• thread

• ReduceAdd

Classes | Public Types | Public Member Functions | Public Attributes | Static Public Attributes | List of all members

cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count > Struct Template Reference

Mixed-precision reduction.

#include <reduction_operators.h>

Collaboration diagram for cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >:

 [legend]

|

Classes

| | struct | Params | | |

|

Public Types

| | using | ElementAccumulator = ElementAccumulator_ | | | | using | Element = Element_ | | | | using | FragmentAccumulator = cutlass::Array< ElementAccumulator, kCount > | | | | using | FragmentElement = cutlass::Array< Element, kCount > | | |

|

Public Member Functions

| | CUTLASS_HOST_DEVICE | ReduceAdd (Params params_=Params()) | | | Constructor. More...
| | | | CUTLASS_HOST_DEVICE FragmentAccumulator | operator() (FragmentAccumulator accumulator, FragmentElement element) const | | | Operator. More...
| | |

|

Public Attributes

| | Params | params | | | Parameters object. More...
| | |

|

Static Public Attributes

| | static int const | kCount = Count | | |

Member Typedef Documentation

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

| using cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::Element = Element_ |

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

| using cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::ElementAccumulator = ElementAccumulator_ |

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

| using cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::FragmentAccumulator = cutlass::Array<ElementAccumulator, kCount> |

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

| using cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::FragmentElement = cutlass::Array<Element, kCount> |

Constructor & Destructor Documentation

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

|

| CUTLASS_HOST_DEVICE cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::ReduceAdd | ( | Params | params_ = Params() | ) | |

| inline |

Member Function Documentation

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

|

| CUTLASS_HOST_DEVICE FragmentAccumulator cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::operator() | ( | FragmentAccumulator | accumulator, | | | | FragmentElement | element | | | ) | | const |

| inline |

Member Data Documentation

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

|

| int const cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::kCount = Count |

| static |

template<typename ElementAccumulator_ , typename Element_ , int Count = 1>

| Params cutlass::reduction::thread::ReduceAdd< ElementAccumulator_, Element_, Count >::params |

---

The documentation for this struct was generated from the following file:

• reduction_operators.h

---

Generated by 1.8.11