allow switching off ducc FFT optimization via #define

mreineck · mreineck · commit 8b4fadfc7a63 · 2024-07-11T19:13:14.000+02:00
diff --git a/src/fft.cpp b/src/fft.cpp
@@ -41,6 +41,9 @@ void do_fft(FINUFFT_PLAN p, CPX *fwBatch) {
     axes.push_back(3);
   }
   ducc0::vfmav<CPX> data(fwBatch, arrdims);
+#ifdef FINUFFT_NO_DUCC0_TWEAKS
+  ducc0::c2c(data, data, axes, p->fftSign < 0, FLT(1), nthreads);
+#else
   if (p->dim == 1)        // 1D: no chance for FFT shortcuts
     ducc0::c2c(data, data, axes, p->fftSign < 0, FLT(1), nthreads);
   else if (p->dim == 2) { // 2D: do partial FFTs
@@ -96,6 +99,7 @@ void do_fft(FINUFFT_PLAN p, CPX *fwBatch) {
       }
     }
   }
+#endif
   delete[] ns;
 #else
   FFTW_EX(p->fftwPlan); // if thisBatchSize<batchSize it wastes some flops

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,9 @@ void do_fft(FINUFFT_PLAN p, CPX *fwBatch) {`
`41`	`41`	`axes.push_back(3);`
`42`	`42`	`}`
`43`	`43`	`ducc0::vfmav<CPX> data(fwBatch, arrdims);`
	`44`	`+#ifdef FINUFFT_NO_DUCC0_TWEAKS`
	`45`	`+ ducc0::c2c(data, data, axes, p->fftSign < 0, FLT(1), nthreads);`
	`46`	`+#else`
`44`	`47`	`if (p->dim == 1) // 1D: no chance for FFT shortcuts`
`45`	`48`	`ducc0::c2c(data, data, axes, p->fftSign < 0, FLT(1), nthreads);`
`46`	`49`	`else if (p->dim == 2) { // 2D: do partial FFTs`
`@@ -96,6 +99,7 @@ void do_fft(FINUFFT_PLAN p, CPX *fwBatch) {`
`96`	`99`	`}`
`97`	`100`	`}`
`98`	`101`	`}`
	`102`	`+#endif`
`99`	`103`	`delete[] ns;`
`100`	`104`	`#else`
`101`	`105`	`FFTW_EX(p->fftwPlan); // if thisBatchSize<batchSize it wastes some flops`