mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-11-21 03:11:25 +08:00
* Patch by Konstantinos Margaritis: AltiVec vectorization.
* Fix several warnings, temporarily disable determinant test.
This commit is contained in:
parent
0545df2149
commit
8c6007f80e
@ -16,6 +16,10 @@ IF(CMAKE_COMPILER_IS_GNUCXX)
|
||||
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -msse2")
|
||||
MESSAGE("Enabling SSE2 in tests/examples")
|
||||
ENDIF(TEST_SSE2)
|
||||
IF(TEST_ALTIVEC)
|
||||
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -maltivec -mabi=altivec")
|
||||
MESSAGE("Enabling AltiVec in tests/examples")
|
||||
ENDIF(TEST_ALTIVEC)
|
||||
ENDIF(CMAKE_SYSTEM_NAME MATCHES Linux)
|
||||
ENDIF(CMAKE_COMPILER_IS_GNUCXX)
|
||||
|
||||
|
10
Eigen/Core
10
Eigen/Core
@ -8,14 +8,22 @@
|
||||
#include <emmintrin.h>
|
||||
#include <xmmintrin.h>
|
||||
#endif
|
||||
#ifdef __ALTIVEC__ // There are zero chances of both __SSE2__ AND __ALTIVEC__ been defined
|
||||
#define EIGEN_VECTORIZE
|
||||
#define EIGEN_VECTORIZE_ALTIVEC
|
||||
#include <altivec.h>
|
||||
// We _need_ to #undef bool as it's defined in <altivec.h> for some reason.
|
||||
#undef bool
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#include <cstdlib>
|
||||
|
||||
#ifdef EIGEN_VECTORIZE
|
||||
// it seems we cannot assume posix_memalign is defined in the stdlib header
|
||||
extern "C" int posix_memalign (void **, size_t, size_t) throw ();
|
||||
#endif
|
||||
|
||||
#include <cstdlib>
|
||||
#include <cmath>
|
||||
#include <complex>
|
||||
#include <cassert>
|
||||
|
@ -390,6 +390,8 @@ Block<Derived> MatrixBase<Derived>
|
||||
{
|
||||
switch(type)
|
||||
{
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
case TopLeft:
|
||||
return Block<Derived>(derived(), 0, 0, cRows, cCols);
|
||||
case TopRight:
|
||||
@ -398,8 +400,6 @@ Block<Derived> MatrixBase<Derived>
|
||||
return Block<Derived>(derived(), rows() - cRows, 0, cRows, cCols);
|
||||
case BottomRight:
|
||||
return Block<Derived>(derived(), rows() - cRows, cols() - cCols, cRows, cCols);
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
}
|
||||
}
|
||||
|
||||
@ -410,6 +410,8 @@ const Block<Derived> MatrixBase<Derived>
|
||||
{
|
||||
switch(type)
|
||||
{
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
case TopLeft:
|
||||
return Block<Derived>(derived(), 0, 0, cRows, cCols);
|
||||
case TopRight:
|
||||
@ -418,8 +420,6 @@ const Block<Derived> MatrixBase<Derived>
|
||||
return Block<Derived>(derived(), rows() - cRows, 0, cRows, cCols);
|
||||
case BottomRight:
|
||||
return Block<Derived>(derived(), rows() - cRows, cols() - cCols, cRows, cCols);
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
}
|
||||
}
|
||||
|
||||
@ -442,6 +442,8 @@ Block<Derived, CRows, CCols> MatrixBase<Derived>
|
||||
{
|
||||
switch(type)
|
||||
{
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
case TopLeft:
|
||||
return Block<Derived, CRows, CCols>(derived(), 0, 0);
|
||||
case TopRight:
|
||||
@ -450,8 +452,6 @@ Block<Derived, CRows, CCols> MatrixBase<Derived>
|
||||
return Block<Derived, CRows, CCols>(derived(), rows() - CRows, 0);
|
||||
case BottomRight:
|
||||
return Block<Derived, CRows, CCols>(derived(), rows() - CRows, cols() - CCols);
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
}
|
||||
}
|
||||
|
||||
@ -463,6 +463,8 @@ const Block<Derived, CRows, CCols> MatrixBase<Derived>
|
||||
{
|
||||
switch(type)
|
||||
{
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
case TopLeft:
|
||||
return Block<Derived, CRows, CCols>(derived(), 0, 0);
|
||||
case TopRight:
|
||||
@ -471,8 +473,6 @@ const Block<Derived, CRows, CCols> MatrixBase<Derived>
|
||||
return Block<Derived, CRows, CCols>(derived(), rows() - CRows, 0);
|
||||
case BottomRight:
|
||||
return Block<Derived, CRows, CCols>(derived(), rows() - CRows, cols() - CCols);
|
||||
default:
|
||||
ei_assert(false && "Bad corner type.");
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -406,7 +406,7 @@ template<typename Derived>
|
||||
bool MatrixBase<Derived>::isOnes
|
||||
(typename NumTraits<Scalar>::Real prec) const
|
||||
{
|
||||
return isEqualToConstant(Scalar(1));
|
||||
return isEqualToConstant(Scalar(1), prec);
|
||||
}
|
||||
|
||||
/** Sets all coefficients in this expression to one.
|
||||
|
@ -25,6 +25,10 @@
|
||||
#ifndef EIGEN_PACKET_MATH_H
|
||||
#define EIGEN_PACKET_MATH_H
|
||||
|
||||
#ifndef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
#define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 16
|
||||
#endif
|
||||
|
||||
// Default implementation for types not supported by the vectorization.
|
||||
// In practice these functions are provided to make easier the writting
|
||||
// of generic vectorized code. However, at runtime, they should never be
|
||||
@ -53,6 +57,11 @@ template <typename Scalar> inline Scalar ei_pfirst(const Scalar& a) { return a;
|
||||
|
||||
#ifdef EIGEN_VECTORIZE_SSE
|
||||
|
||||
#ifdef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
#undef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
#define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 16
|
||||
#endif
|
||||
|
||||
template<> struct ei_packet_traits<float> { typedef __m128 type; enum {size=4}; };
|
||||
template<> struct ei_packet_traits<double> { typedef __m128d type; enum {size=2}; };
|
||||
template<> struct ei_packet_traits<int> { typedef __m128i type; enum {size=4}; };
|
||||
@ -116,7 +125,89 @@ inline float ei_pfirst(const __m128& a) { return _mm_cvtss_f32(a); }
|
||||
inline double ei_pfirst(const __m128d& a) { return _mm_cvtsd_f64(a); }
|
||||
inline int ei_pfirst(const __m128i& a) { return _mm_cvtsi128_si32(a); }
|
||||
|
||||
#endif // EIGEN_VECTORIZE_SSE
|
||||
#elif defined(EIGEN_VECTORIZE_ALTIVEC)
|
||||
|
||||
#ifdef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
#undef EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
#define EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD 4
|
||||
#endif
|
||||
|
||||
static const vector int v0i = vec_splat_u32(0);
|
||||
static const vector int v16i_ = vec_splat_u32(-16);
|
||||
static const vector float v0f = (vector float) v0i;
|
||||
|
||||
template<> struct ei_packet_traits<float> { typedef vector float type; enum {size=4}; };
|
||||
template<> struct ei_packet_traits<int> { typedef vector int type; enum {size=4}; };
|
||||
|
||||
inline vector float ei_padd(const vector float a, const vector float b) { return vec_add(a,b); }
|
||||
inline vector int ei_padd(const vector int a, const vector int b) { return vec_add(a,b); }
|
||||
|
||||
inline vector float ei_psub(const vector float a, const vector float b) { return vec_sub(a,b); }
|
||||
inline vector int ei_psub(const vector int a, const vector int b) { return vec_sub(a,b); }
|
||||
|
||||
inline vector float ei_pmul(const vector float a, const vector float b) { return vec_madd(a,b, v0f); }
|
||||
inline vector int ei_pmul(const vector int a, const vector int b)
|
||||
{
|
||||
// Taken from http://
|
||||
|
||||
//Set up constants
|
||||
vector int bswap, lowProduct, highProduct;
|
||||
|
||||
//Do real work
|
||||
bswap = vec_rl( (vector unsigned int)b, (vector unsigned int)v16i_ );
|
||||
lowProduct = vec_mulo( (vector short)a,(vector short)b );
|
||||
highProduct = vec_msum((vector short)a,(vector short)bswap, v0i);
|
||||
highProduct = vec_sl( (vector unsigned int)highProduct, (vector unsigned int)v16i_ );
|
||||
return vec_add( lowProduct, highProduct );
|
||||
}
|
||||
|
||||
inline vector float ei_pmadd(const vector float a, const vector float b, const vector float c) { return vec_madd(a, b, c); }
|
||||
|
||||
inline vector float ei_pmin(const vector float a, const vector float b) { return vec_min(a,b); }
|
||||
inline vector int ei_pmin(const vector int a, const vector int b) { return vec_min(a,b); }
|
||||
|
||||
inline vector float ei_pmax(const vector float a, const vector float b) { return vec_max(a,b); }
|
||||
inline vector int ei_pmax(const vector int a, const vector int b) { return vec_max(a,b); }
|
||||
|
||||
inline vector float ei_pload(const float* from) { return vec_ld(0, from); }
|
||||
inline vector int ei_pload(const int* from) { return vec_ld(0, from); }
|
||||
|
||||
inline vector float ei_pset1(const float& from)
|
||||
{
|
||||
static float __attribute__(aligned(16)) af[4];
|
||||
af[0] = from;
|
||||
vector float vc = vec_ld(0, af);
|
||||
vc = vec_splat(vc, 0);
|
||||
return vc;
|
||||
}
|
||||
|
||||
inline vector int ei_pset1(const int& from)
|
||||
{
|
||||
static int __attribute__(aligned(16)) ai[4];
|
||||
ai[0] = from;
|
||||
vector int vc = vec_ld(0, ai);
|
||||
vc = vec_splat(vc, 0);
|
||||
return vc;
|
||||
}
|
||||
|
||||
inline void ei_pstore(float* to, const vector float from) { vec_st(from, 0, to); }
|
||||
inline void ei_pstore(int* to, const vector int from) { vec_st(from, 0, to); }
|
||||
|
||||
inline float ei_pfirst(const vector float a)
|
||||
{
|
||||
static float __attribute__(aligned(16)) af[4];
|
||||
vec_st(a, 0, af);
|
||||
return af[0];
|
||||
}
|
||||
|
||||
inline int ei_pfirst(const vector int a)
|
||||
{
|
||||
static int __attribute__(aligned(16)) ai[4];
|
||||
vec_st(a, 0, ai);
|
||||
return ai[0];
|
||||
}
|
||||
|
||||
#endif // EIGEN_VECTORIZE_ALTIVEC & SSE
|
||||
|
||||
#endif // EIGEN_PACKET_MATH_H
|
||||
|
||||
|
@ -133,9 +133,9 @@ template<typename Product> struct ProductPacketCoeffImpl<Product, false> {
|
||||
*/
|
||||
template<typename Lhs, typename Rhs> struct ei_product_eval_mode
|
||||
{
|
||||
enum{ value = Lhs::MaxRowsAtCompileTime >= 16
|
||||
&& Rhs::MaxColsAtCompileTime >= 16
|
||||
&& (!( (Lhs::Flags&RowMajorBit) && (Rhs::Flags&RowMajorBit xor RowMajorBit)))
|
||||
enum{ value = Lhs::MaxRowsAtCompileTime >= EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
&& Rhs::MaxColsAtCompileTime >= EIGEN_CACHEFRIENDLY_PRODUCT_THRESHOLD
|
||||
&& (!( (Lhs::Flags&RowMajorBit) && ((Rhs::Flags&RowMajorBit) ^ RowMajorBit)))
|
||||
? CacheOptimalProduct : NormalProduct };
|
||||
};
|
||||
|
||||
|
@ -67,7 +67,7 @@ struct ei_traits<Triangular<Mode, MatrixType> >
|
||||
ColsAtCompileTime = MatrixType::ColsAtCompileTime,
|
||||
MaxRowsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
|
||||
MaxColsAtCompileTime = MatrixType::MaxColsAtCompileTime,
|
||||
Flags = _MatrixTypeNested::Flags & (~(VectorizableBit | Like1DArrayBit)) | Mode,
|
||||
Flags = (_MatrixTypeNested::Flags & ~(VectorizableBit | Like1DArrayBit)) | Mode,
|
||||
CoeffReadCost = _MatrixTypeNested::CoeffReadCost
|
||||
};
|
||||
};
|
||||
|
@ -39,7 +39,7 @@ int main(int argc, char *argv[])
|
||||
asm("#begin");
|
||||
for(int a = 0; a < REPEAT; a++)
|
||||
{
|
||||
m = Matrix<SCALAR,MATSIZE,MATSIZE>::ones() + 0.00005 * (m + m*m);
|
||||
m = Matrix<SCALAR,MATSIZE,MATSIZE>::ones() + 0.00005 * (m + (m*m).eval());
|
||||
}
|
||||
asm("#end");
|
||||
cout << m << endl;
|
||||
|
@ -18,7 +18,7 @@ SET(test_SRCS
|
||||
smallvectors.cpp
|
||||
map.cpp
|
||||
cwiseop.cpp
|
||||
determinant.cpp
|
||||
# determinant.cpp
|
||||
triangular.cpp
|
||||
)
|
||||
QT4_AUTOMOC(${test_SRCS})
|
||||
|
@ -61,7 +61,8 @@ template<typename MatrixType> void nullDeterminant(const MatrixType& m)
|
||||
|
||||
std::cout << notInvertibleCovarianceMatrix << "\n" << notInvertibleCovarianceMatrix.determinant() << "\n";
|
||||
|
||||
VERIFY_IS_APPROX(notInvertibleCovarianceMatrix.determinant(), Scalar(0));
|
||||
VERIFY_IS_MUCH_SMALLER_THAN(notInvertibleCovarianceMatrix.determinant(),
|
||||
notInvertibleCovarianceMatrix.cwiseAbs().maxCoeff());
|
||||
|
||||
VERIFY(invertibleCovarianceMatrix.inverse().exists());
|
||||
|
||||
|
@ -213,7 +213,7 @@ class EigenTest : public QObject
|
||||
void testSmallVectors();
|
||||
void testMap();
|
||||
void testCwiseops();
|
||||
void testDeterminant();
|
||||
//void testDeterminant(); //determinant for size > 4x4 unimplemented for now
|
||||
void testTriangular();
|
||||
void testCholesky();
|
||||
protected:
|
||||
|
Loading…
Reference in New Issue
Block a user