GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 03:39:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

port Jacobi to new ei_pset1/ei_pload API

Browse Source
This commit is contained in:
Gael Guennebaud 2010-07-19 16:51:38 +02:00
parent c2ee454df4
commit 1ed4233fd2
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Eigen/src/Jacobi/Jacobi.h
View File

@ -322,8 +322,8 @@ void /*EIGEN_DONT_INLINE*/ ei_apply_rotation_in_the_plane(VectorX& _x, VectorY&
Index alignedStart = ei_first_aligned(y, size); Index alignedStart = ei_first_aligned(y, size);
Index alignedEnd = alignedStart + ((size-alignedStart)/PacketSize)*PacketSize; Index alignedEnd = alignedStart + ((size-alignedStart)/PacketSize)*PacketSize;
const Packet pc = ei_pset1(Scalar(j.c())); const Packet pc = ei_pset1<Packet>(j.c());
const Packet ps = ei_pset1(Scalar(j.s())); const Packet ps = ei_pset1<Packet>(j.s());
ei_conj_helper<Packet,Packet,NumTraits<Scalar>::IsComplex,false> pcj; ei_conj_helper<Packet,Packet,NumTraits<Scalar>::IsComplex,false> pcj;
for(Index i=0; i<alignedStart; ++i) for(Index i=0; i<alignedStart; ++i)
@ -341,8 +341,8 @@ void /*EIGEN_DONT_INLINE*/ ei_apply_rotation_in_the_plane(VectorX& _x, VectorY&
{ {
for(Index i=alignedStart; i<alignedEnd; i+=PacketSize) for(Index i=alignedStart; i<alignedEnd; i+=PacketSize)
{ {
Packet xi = ei_pload(px); Packet xi = ei_pload<Packet>(px);
Packet yi = ei_pload(py); Packet yi = ei_pload<Packet>(py);
ei_pstore(px, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi))); ei_pstore(px, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi)));
ei_pstore(py, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi))); ei_pstore(py, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi)));
px += PacketSize; px += PacketSize;
@ -354,10 +354,10 @@ void /*EIGEN_DONT_INLINE*/ ei_apply_rotation_in_the_plane(VectorX& _x, VectorY&
Index peelingEnd = alignedStart + ((size-alignedStart)/(Peeling*PacketSize))*(Peeling*PacketSize); Index peelingEnd = alignedStart + ((size-alignedStart)/(Peeling*PacketSize))*(Peeling*PacketSize);
for(Index i=alignedStart; i<peelingEnd; i+=Peeling*PacketSize) for(Index i=alignedStart; i<peelingEnd; i+=Peeling*PacketSize)
{ {
Packet xi = ei_ploadu(px); Packet xi = ei_ploadu<Packet>(px);
Packet xi1 = ei_ploadu(px+PacketSize); Packet xi1 = ei_ploadu<Packet>(px+PacketSize);
Packet yi = ei_pload (py); Packet yi = ei_pload <Packet>(py);
Packet yi1 = ei_pload (py+PacketSize); Packet yi1 = ei_pload <Packet>(py+PacketSize);
ei_pstoreu(px, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi))); ei_pstoreu(px, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi)));
ei_pstoreu(px+PacketSize, ei_padd(ei_pmul(pc,xi1),pcj.pmul(ps,yi1))); ei_pstoreu(px+PacketSize, ei_padd(ei_pmul(pc,xi1),pcj.pmul(ps,yi1)));
ei_pstore (py, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi))); ei_pstore (py, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi)));
@ -367,8 +367,8 @@ void /*EIGEN_DONT_INLINE*/ ei_apply_rotation_in_the_plane(VectorX& _x, VectorY&
} }
if(alignedEnd!=peelingEnd) if(alignedEnd!=peelingEnd)
{ {
Packet xi = ei_ploadu(x+peelingEnd); Packet xi = ei_ploadu<Packet>(x+peelingEnd);
Packet yi = ei_pload (y+peelingEnd); Packet yi = ei_pload <Packet>(y+peelingEnd);
ei_pstoreu(x+peelingEnd, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi))); ei_pstoreu(x+peelingEnd, ei_padd(ei_pmul(pc,xi),pcj.pmul(ps,yi)));
ei_pstore (y+peelingEnd, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi))); ei_pstore (y+peelingEnd, ei_psub(ei_pmul(pc,yi),ei_pmul(ps,xi)));
} }