Structure to compute the matrix-vector product using SIMT math instructions.
#include <gemv.h>
|
| using | Shape = typename Core_::Shape |
| |
| using | Operator = typename Core_::Operator |
| | The MMA operator that computes GEMV. More...
|
| |
| using | IteratorA = typename Core_::IteratorA |
| | Iterates over A in global memory. More...
|
| |
| using | IteratorB = typename Core_::IteratorB |
| | Iterates over B in global memory. More...
|
| |
| using | IteratorC = typename Core_::IteratorC |
| | Fragment of operand C loaded from global memory. More...
|
| |
| using | FragmentA = typename IteratorA::Fragment |
| | Fragment of operand A loaded from global memory. More...
|
| |
| using | FragmentB = typename IteratorB::Fragment |
| | Fragment of operand B loaded from global memory. More...
|
| |
| using | FragmentC = typename Operator::FragmentC |
| | Fragment of operand accumulator loaded/stored to global memory. More...
|
| |
| using | ThreadShape = typename Core_::ThreadShape |
| | Shape of the per-thread GEMV operation. More...
|
| |
< source accumualtor tile
- Parameters
-
| problem_size | problem size of batched GEMV |
| accum | destination accumulator tile |
| iterator_A | iterator over A operand in global memory |
| iterator_B | iterator over B operand in global memory |
The documentation for this class was generated from the following file: