Commit bf2914d9 authored by Luca Tornatore's avatar Luca Tornatore
Browse files

polishing 2

parent 03d87005

fourier_transform.c

+25 −8
Original line number Diff line number Diff line
@@ -2,9 +2,14 @@ 
#include "allvars.h"

#include "proto.h"



void fftw_data(){



 #if defined(USE_FFTW) && !defined(CUFFTMP)

                                                // ------------------------------------

#if defined(USE_FFTW) && !defined(CUFFTMP)      //  PERFORM FFT on CPU with FFTW

						// ------------------------------------ 



void fftw_data ( void )

{



  // FFT transform the data (using distributed FFTW)

  if(rank == 0)printf("PERFORMING FFT\n");
@@ -81,9 +86,18 @@ void fftw_data(){ 


  timing_wt.fftw += CPU_TIME_wt - start;



 #endif

  return;

}

                                         // ------------------------------------

#else                                    //  PERFORM FFT ON GPU USING CUFFTMP

					 // ------------------------------------



void fftw_data ( void )

{



  // FFT transform the data (using distributed FFTW)

  if(rank == 0)printf("PERFORMING FFT\n");



 #if defined(USE_FFTW) && defined(CUFFTMP)

  // FFT transform the data using cuFFT                                                                                                    

  if(rank==0)printf("PERFORMING CUDA FFT\n");
@@ -105,10 +119,13 @@ void fftw_data(){ 
 

  timing_wt.cufftmp += CPU_TIME_wt - start;



 #endif

  return;

}





#endif                                  // END OF FFT SELECTION



   

}



void write_fftw_data(){