added simd capability

This commit is contained in:
Mark Borgerding
2005-06-24 03:03:31 +00:00
parent 24be1c5850
commit 7f6cbeab2e
10 changed files with 112 additions and 9 deletions

View File

@ -6,6 +6,8 @@
#include "pstats.h"
#define CHK fprintf(stderr,"line %d\n" , __LINE__ )
int main(int argc,char ** argv)
{
int nfft=1024;
@ -31,20 +33,41 @@ int main(int argc,char ** argv)
break;
}
}
CHK;
#ifdef USE_SIMD
buf=(kiss_fft_cpx*)memalign(sizeof(kiss_fft_cpx),sizeof(kiss_fft_cpx) * nfft);
bufout=(kiss_fft_cpx*)memalign(sizeof(kiss_fft_cpx),sizeof(kiss_fft_cpx) * nfft);
numffts /= 4;
fprintf(stderr,"since SIMD implementation does 4 ffts at a time, numffts is being reduced to %d\n",numffts);
#else
buf=(kiss_fft_cpx*)malloc(sizeof(kiss_fft_cpx) * nfft);
bufout=(kiss_fft_cpx*)malloc(sizeof(kiss_fft_cpx) * nfft);
#endif
fprintf(stderr,"buf at %p, bufout at %p\n",buf,bufout);
CHK;
for (i=0;i<nfft;++i ) {
#ifdef USE_SIMD
buf[i].r = _mm_set_ps1((float)( rand() - RAND_MAX/2));
buf[i].i = _mm_set_ps1((float)( rand() - RAND_MAX/2));
#else
buf[i].r = rand() - RAND_MAX/2;
buf[i].i = rand() - RAND_MAX/2;
#endif
}
CHK;
pstats_init();
CHK;
st = kiss_fft_alloc( nfft ,isinverse ,0,0);
CHK;
for (i=0;i<numffts;++i)
kiss_fft( st ,buf,bufout );
CHK;
free(st);