fixed MagneX.H header for compiling issue

Author: tyh0123

Source/MagneX.H

@@ -1,3 +1,13 @@
+#ifdef AMREX_USE_CUDA
+#include <cufft.h>
+#include <torch/script.h>
+#else
+#include <fftw3.h>
+#ifdef AMREX_USE_MPI
+#include <fftw3-mpi.h>
+#endif
+#endif
+
 #include <AMReX_MultiFab.H>
 
 #include "MagneX_namespace.H"
@@ -176,7 +186,8 @@ void WritePlotfile(MultiFab& Ms,
                    const Real& time,
                    const int& plt_step);
 
-#ifdef AMREX_USE_ML
+
+                        
 
 at::Tensor PackMfieldToTensorDynamic(
   const amrex::Array<amrex::MultiFab, AMREX_SPACEDIM>& Mfield,
@@ -209,6 +220,4 @@ void MoveModuleToDevice(torch::jit::script::Module& m,
                       const torch::Device& device);
 
 // Read expected_spatial = [nx, ny, nz] from normalizer module
-amrex::IntVect GetExpectedSpatial(torch::jit::script::Module& x_norm_module);
-
-#endif
+amrex::IntVect GetExpectedSpatial(torch::jit::script::Module& x_norm_module);