usyurk.tcc (ulmblas/impl/level3/ukernel/usyurk.tcc)

      1
      2
      3
      4
      5
      6
      7
      8
      9
     10
     11
     12
     13
     14
     15
     16
     17
     18
     19
     20
     21
     22
     23
     24
     25
     26
     27
     28
     29
     30
     31
     32
     33
     34
     35
     36
     37
     38
     39
     40
     41
     42
     43
     44
     45
     46
     47
     48
     49
     50
     51
     52
     53
     54

#ifndef ULMBLAS_IMPL_LEVEL3_UKERNEL_USYURK_TCC

#define ULMBLAS_IMPL_LEVEL3_UKERNEL_USYURK_TCC 1

#include <ulmblas/impl/level1extensions/truaxpy.h>

#include <ulmblas/impl/level1extensions/truscal.h>

#include <ulmblas/impl/level3/ukernel/ugemm.h>

#include <ulmblas/impl/level3/ukernel/usyurk.h>

namespace ulmBLAS {

template <typename IndexType, typename T, typename Beta, typename TC>

void

usyurk(IndexType    mr,

       IndexType    nr,

       IndexType    kc,

       IndexType    ic,

       IndexType    jc,

       const T      &alpha,

       const T      *A,

       const T      *B,

       const Beta   &beta,

       TC           *C,

       IndexType    incRowC,

       IndexType    incColC,

       const T      *nextA,

       const T      *nextB)

{

    const IndexType MR = BlockSizeUGemm<T>::MR;

    const IndexType NR = BlockSizeUGemm<T>::NR;

    T   C_[MR*NR];

    ugemm(kc, alpha, A, B, T(0), C_, IndexType(1), MR, nextA, nextB);

    if (jc>ic) {

        gescal(jc-ic, nr, beta, C, incRowC, incColC);

        geaxpy(jc-ic, nr, Beta(1), C_, IndexType(1), MR, C, incRowC, incColC);

        truscal(mr-(jc-ic), nr, false, beta,

                &C[(jc-ic)*incRowC], incRowC, incColC);

        truaxpy(mr-(jc-ic), nr, false, Beta(1),

                &C_[jc-ic], IndexType(1), MR,

                &C[(jc-ic)*incRowC], incRowC, incColC);

    } else {

        truscal(mr, nr-(ic-jc), false, beta,

                &C[(ic-jc)*incColC], incRowC, incColC);

        truaxpy(mr, nr-(ic-jc), false, Beta(1),

                &C_[(ic-jc)*MR], IndexType(1), MR,

                &C[(ic-jc)*incColC], incRowC, incColC);

   }

}

} // namespace ulmBLAS

#endif // ULMBLAS_IMPL_LEVEL3_UKERNEL_USYURK_TCC