mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-04 18:54:00 +08:00
28539e7597
the modifications to initial code follow: * changed build system from plain makefiles to cmake * added eigen2 (4 versions: vec/novec and fixed/dynamic), GMM++, MTL4 interfaces * added "transposed matrix * vector" product action * updated blitz interface to use condensed products instead of hand coded loops * removed some deprecated interfaces * changed default storage order to column major for all libraries * new generic bench timer strategy which is supposed to be more accurate * various code clean-up
119 lines
2.8 KiB
C++
119 lines
2.8 KiB
C++
//=====================================================
|
|
// File : ATLAS_LU_solve_interface.hh
|
|
// Author : L. Plagne <laurent.plagne@edf.fr)>
|
|
// Copyright (C) EDF R&D, lun sep 30 14:23:22 CEST 2002
|
|
//=====================================================
|
|
//
|
|
// This program 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 2
|
|
// of the License, or (at your option) any later version.
|
|
//
|
|
// This program 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 this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
//
|
|
#ifndef ATLAS_LU_solve_interface_HH
|
|
#define ATLAS_LU_solve_interface_HH
|
|
#include "ATLAS_interface.hh"
|
|
extern "C"
|
|
{
|
|
#include <atlas_level1.h>
|
|
#include <atlas_level2.h>
|
|
#include <atlas_level3.h>
|
|
#include "cblas.h"
|
|
#include <atlas_lapack.h>
|
|
|
|
}
|
|
|
|
template<class real>
|
|
class ATLAS_LU_solve_interface : public ATLAS_interface<real>
|
|
{
|
|
public :
|
|
|
|
typedef typename ATLAS_interface<real>::gene_matrix gene_matrix;
|
|
typedef typename ATLAS_interface<real>::gene_vector gene_vector;
|
|
|
|
typedef int * Pivot_Vector;
|
|
|
|
inline static void new_Pivot_Vector(Pivot_Vector & pivot, int N)
|
|
{
|
|
|
|
pivot = new int[N];
|
|
|
|
}
|
|
|
|
inline static void free_Pivot_Vector(Pivot_Vector & pivot)
|
|
{
|
|
|
|
delete pivot;
|
|
|
|
}
|
|
|
|
|
|
inline static void LU_factor(gene_matrix & LU, Pivot_Vector & pivot, int N)
|
|
{
|
|
|
|
int error=ATL_dgetrf(CblasColMajor,N,N,LU,N,pivot);
|
|
|
|
}
|
|
|
|
inline static void LU_solve(const gene_matrix & LU, const Pivot_Vector pivot, const gene_vector &B, gene_vector X, int N)
|
|
{
|
|
|
|
copy_vector(B,X,N);
|
|
ATL_dgetrs(CblasColMajor,CblasNoTrans,N,1,LU,N,pivot,X,N);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
template<>
|
|
class ATLAS_LU_solve_interface<float> : public ATLAS_interface<float>
|
|
{
|
|
public :
|
|
|
|
typedef int * Pivot_Vector;
|
|
|
|
inline static void new_Pivot_Vector(Pivot_Vector & pivot, int N)
|
|
{
|
|
|
|
pivot = new int[N];
|
|
|
|
}
|
|
|
|
inline static void free_Pivot_Vector(Pivot_Vector & pivot)
|
|
{
|
|
|
|
delete pivot;
|
|
|
|
}
|
|
|
|
|
|
inline static void LU_factor(gene_matrix & LU, Pivot_Vector & pivot, int N)
|
|
{
|
|
|
|
int error=ATL_sgetrf(CblasColMajor,N,N,LU,N,pivot);
|
|
|
|
}
|
|
|
|
inline static void LU_solve(const gene_matrix & LU, const Pivot_Vector pivot, const gene_vector &B, gene_vector X, int N)
|
|
{
|
|
|
|
copy_vector(B,X,N);
|
|
ATL_sgetrs(CblasColMajor,CblasNoTrans,N,1,LU,N,pivot,X,N);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|