42 namespace threadblock {
    52   using Shape = 
typename Core_::Shape;
    99     iterator_A.load(frag_A);
   100     iterator_B.load(frag_B);
   108     int gemm_k = problem_size.
k();
   110     if (gemm_k < Shape::kK)
   112       iterator_A.clear_mask();
   113       iterator_B.clear_mask();
   118     for (; gemm_k > 0; gemm_k -= Shape::kK) {
   119       thread_mma(accum, frag_A, frag_B, accum);
   121       iterator_A.load(frag_A);
   122       iterator_B.load(frag_B);
   126       if (gemm_k < Shape::kK)
   128         iterator_A.clear_mask();
   129         iterator_B.clear_mask();
 Definition: aligned_buffer.h:35
CUTLASS_DEVICE Gemv()
Definition: gemv.h:80
typename IteratorB::Fragment FragmentB
Fragment of operand B loaded from global memory. 
Definition: gemv.h:70
Definition: include/cutlass/gemm/gemm.h:94
Defines common types used for all GEMM-like operators. 
typename Core_::Operator Operator
The MMA operator that computes GEMV. 
Definition: gemv.h:55
CUTLASS_HOST_DEVICE Index const & k() const 
Returns the GEMM K coordinate. 
Definition: include/cutlass/gemm/gemm.h:145
Statically sized array of elements that accommodates all CUTLASS-supported numeric types and is safe ...
Structure to compute the matrix-vector product using SIMT math instructions. 
Definition: gemv.h:50
Defines a Shape template for matrix tiles. 
Top-level include for all CUTLASS numeric types. 
typename Core_::ThreadShape ThreadShape
Shape of the per-thread GEMV operation. 
Definition: gemv.h:76
typename Core_::IteratorA IteratorA
Iterates over A in global memory. 
Definition: gemv.h:58
typename Operator::FragmentC FragmentC
Fragment of operand accumulator loaded/stored to global memory. 
Definition: gemv.h:73
typename Core_::IteratorB IteratorB
Iterates over B in global memory. 
Definition: gemv.h:61
#define CUTLASS_GEMM_LOOP
Definition: cutlass.h:112
typename IteratorA::Fragment FragmentA
Fragment of operand A loaded from global memory. 
Definition: gemv.h:67
CUTLASS_DEVICE void operator()(GemmCoord const &problem_size, FragmentC &accum, IteratorA iterator_A, IteratorB iterator_B, FragmentC const &src_accum)
Definition: gemv.h:83
Basic include for CUTLASS. 
typename Core_::Shape Shape
Definition: gemv.h:52
typename Core_::IteratorC IteratorC
Fragment of operand C loaded from global memory. 
Definition: gemv.h:64