schurgemm.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_MULTIFRONTAL_LU_SCHURGEMMA_H
#define MYRAMATH_MULTIFRONTAL_LU_SCHURGEMMA_H

#include <myramath/multifrontal/detail/schurgemm.h>

namespace myra {

// Forward declarations.
template<class Number> class LUKernel;

namespace multifrontal {
namespace detail {

// Forwarde declarations.
template<class Number> class XContainer;

namespace lu {

// Forward declarations.
template<class Kernel> class LUContainer;

namespace schurgemm {

template<class Number> class JobGraphBase : public ::myra::multifrontal::detail::schurgemm::JobGraphBase2<Number>
  {
  public:

    // Structural typedefs for base class.
    typedef ::myra::multifrontal::detail::schurgemm::JobGraphBase2<Number> Base;

    // Typedefs related to numeric storage.
    typedef ::myra::LUKernel<Number> Kernel;
    typedef ::myra::multifrontal::detail::lu::LUContainer<Kernel> LUContainer;
    typedef ::myra::multifrontal::detail::XContainer<Number> XContainer;
    typedef CMatrixRange<Number> CRange;
    typedef MatrixRange<Number> Range;
    
    // Constructor - requires const references to X and Y, mutable reference to S
    JobGraphBase(const LUContainer* in_LU, const XContainer* in_X, const XContainer* in_Y, const MatrixRange<Number>& in_S)
      : Base(&in_LU->tree(), in_X, in_Y, in_S), X(in_X), Y(in_Y), S(in_S) { }

    // Virtual copy constructor.
    virtual ::myra::JobGraphBase* clone() const
      { return new JobGraphBase(*this); }

  protected:

    // Updates Sn(i,j) += Xn(k,i)'*Yn(k,j)
    virtual uint64_t update(int n, int k, int i, int j)
      {
      Number one(1);
      Number zero(0);
      const Range Sn_ij = this->s(n,i,j);
      const CRange Xn_ki = this->x(n,k,i);
      const CRange Yn_kj = this->y(n,k,j);
      return gemm_nwork(Sn_ij, Xn_ki, 'T', Yn_kj, 'N', one, one);
      }

  private:

    // Reference to underlying X and Y (const)
    const XContainer* X;
    const XContainer* Y;

    // Output range.
    const MatrixRange<Number> S;

  }; // class JobGraph

} } } } } // namespace

#endif