CuHyb.hpp

/*
 * Copyright (c) 2007-2014, A. N. Yzelman,   Utrecht University 2007-2011;
 *                                                    KU Leuven 2011-2014.
 *                          R. H. Bisseling, Utrecht University 2007-2014.
 * 
 * This file is part of the Sparse Library.
 * 
 * This library was developed under supervision of Prof. dr. Rob H. Bisseling at
 * Utrecht University, from 2007 until 2011. From 2011-2014, development continued 
 * at KU Leuven, where Prof. dr. Dirk Roose contributed significantly to the ideas 
 * behind the newer parts of the library code.
 * 
 *     The Sparse Library is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by the
 *     Free Software Foundation, either version 3 of the License, or (at your
 *     option) any later version.
 * 
 *     The Sparse Library is distributed in the hope that it will be useful, but
 *     WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY 
 *     or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 *     for more details.
 * 
 *     You should have received a copy of the GNU General Public License along
 *     with the Sparse Library. If not, see <http://www.gnu.org/licenses/>.
 */

/*
 * File created by:
 *     A. N. Yzelman, Dept. of Computer Science, KU Leuven, 2014.
 */


#include <assert.h>

#include <cuda_runtime.h>
#include <cusparse_v2.h>

#include "SparseMatrix.hpp"

#ifndef _H_CUHYB
#define _H_CUHYB

class CuHyb : public SparseMatrix< double, size_t > {

  private:

  protected:

        cusparseHandle_t handle;

        cusparseMatDescr_t descrA;

        cusparseHybMat_t hybA;

        double * GPUx;

        double * GPUz;

        static int compareTriplets( const void * left, const void * right );

        size_t i;

        size_t j;


  public:

        CuHyb();

        CuHyb( std::string file, double zero = 0 );
        
        CuHyb( const ULI number_of_nonzeros, const ULI number_of_rows, const ULI number_of_cols, double zero );

        CuHyb( std::vector< Triplet< double > > input, ULI m, ULI n, double zero );

        virtual ~CuHyb();

        virtual void load( std::vector< Triplet< double > >& input, ULI m, ULI n, double zero );

        virtual void zxa( const double *__restrict__ x, double *__restrict__ z );

        virtual void zax( const double *__restrict__ x, double *__restrict__ z );

        virtual void zax( const double *__restrict__ x, double *__restrict__ z, const unsigned long int repeat, const clockid_t clock_id = 0, double *elapsed_time = NULL );

        virtual size_t bytesUsed();

        virtual void getFirstIndexPair( size_t &i, size_t &j );

};

#endif