do reductions on MFab and not BaseFab

AMReX-FHD · Nov 5, 2023 · a7d9004 · a7d9004
1 parent 37bd3c3
commit a7d9004
Show file tree

Hide file tree

Showing 2 changed files with 350 additions and 214 deletions.
diff --git a/src_analysis/TurbSpectra.H b/src_analysis/TurbSpectra.H
@@ -47,17 +47,25 @@ void Assert_rocfft_status (std::string const& name, rocfft_status status);
 #endif
 
 #if defined(HEFFTE_FFTW) || defined(HEFFTE_CUFFT) || defined(HEFFTE_ROCFFT)
-void IntegrateKScalarHeffte(const BaseFab<GpuComplex<Real> >& spectral_field,
-                      	    const std::string& name, const Real& scaling,
-                      	    const Box& c_local_box,
-                      	    const Real& sqrtnpts,
-                      	    const int& step);
-void IntegrateKVelocityHeffte(const BaseFab<GpuComplex<Real> >& spectral_fieldx,
-			      const BaseFab<GpuComplex<Real> >& spectral_fieldy,
-			      const BaseFab<GpuComplex<Real> >& spectral_fieldz,
-                              const std::string& name, const Real& scaling,
-                              const Box& c_local_box,
-                              const int& step);
+void IntegrateKScalarHeffte(const MultiFab& cov_mag,
+                      	    const std::string& name,
+                            const int& step,
+                            const int& comp);
+//void IntegrateKScalarHeffte(const BaseFab<GpuComplex<Real> >& spectral_field,
+//                      	    const std::string& name, const Real& scaling,
+//                      	    const Box& c_local_box,
+//                      	    const Real& sqrtnpts,
+//                      	    const int& step);
+void IntegrateKVelocityHeffte(const MultiFab& cov_mag,
+                              const std::string& name,
+                              const int& step,
+                              const int& comp);
+//void IntegrateKVelocityHeffte(const BaseFab<GpuComplex<Real> >& spectral_fieldx,
+//			      const BaseFab<GpuComplex<Real> >& spectral_fieldy,
+//			      const BaseFab<GpuComplex<Real> >& spectral_fieldz,
+//                              const std::string& name, const Real& scaling,
+//                              const Box& c_local_box,
+//                              const int& step);
 void TurbSpectrumScalarHeffte(const MultiFab& variables, 
                               const amrex::Geometry& geom, 
                               const int& step,