CUTLASS: cutlass::reduction::BatchedReduction< BatchedReductionTraits_ > 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

• BatchedReduction

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

cutlass::reduction::BatchedReduction< BatchedReductionTraits_ > Struct Template Reference

#include <batched_reduction.h>

|

Public Types

| | typedef BatchedReduction< BatchedReductionTraits_ > | This_ | | | This class. More...
| | | | typedef BatchedReductionTraits_ | Traits | | | The traits. More...
| | | | typedef Traits::Params | Params | | | Params. More...
| | | | typedef Traits::Functor | Functor | | | functor More...
| | |

|

Public Member Functions

| | CUTLASS_DEVICE | BatchedReduction (Params const &params_) | | | ctor More...
| | | | CUTLASS_DEVICE void | run () | | | | template<bool ThreadShapeMultiple2> | | CUTLASS_DEVICE void | functor_caller (typename Traits::ScalarAccum const *accum, typename Traits::ScalarAccum const *old, typename Traits::ScalarAccum *output) | | |

|

Static Public Member Functions

| | static __host__ cudaError_t | launch (Params const &params, cudaStream_t stream=cudaStreamDefault) | | | Launch the kernel. More...
| | |

|

Public Attributes

| | Params const & | params | | | The params. More...
| | | | Functor | functor | | |

Member Typedef Documentation

template<typename BatchedReductionTraits_ >

| typedef Traits::Functor cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::Functor |

template<typename BatchedReductionTraits_ >

| typedef Traits::Params cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::Params |

template<typename BatchedReductionTraits_ >

| typedef BatchedReduction<BatchedReductionTraits_> cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::This_ |

template<typename BatchedReductionTraits_ >

| typedef BatchedReductionTraits_ cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::Traits |

Constructor & Destructor Documentation

template<typename BatchedReductionTraits_ >

|

| CUTLASS_DEVICE cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::BatchedReduction | ( | Params const & | params_ | ) | |

| inline |

Member Function Documentation

template<typename BatchedReductionTraits_ >

template<bool ThreadShapeMultiple2>

|

| CUTLASS_DEVICE void cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::functor_caller | ( | typename Traits::ScalarAccum const * | accum, | | | | typename Traits::ScalarAccum const * | old, | | | | typename Traits::ScalarAccum * | output | | | ) | | |

| inline |

template<typename BatchedReductionTraits_ >

|

| static __host__ cudaError_t cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::launch | ( | Params const & | params, | | | | cudaStream_t | stream = cudaStreamDefault | | | ) | | |

| inlinestatic |

template<typename BatchedReductionTraits_ >

|

| CUTLASS_DEVICE void cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::run | ( | | ) | |

| inline |

main operation method D = alpha * Reduction(A) + beta * C

Member Data Documentation

template<typename BatchedReductionTraits_ >

| Functor cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::functor |

template<typename BatchedReductionTraits_ >

| Params const& cutlass::reduction::BatchedReduction< BatchedReductionTraits_ >::params |

---

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

• batched_reduction.h

---

Generated by 1.8.11