Stabilize EScalar CUDA fallback path

Rewritten done by Codex.
This still has errors, do not pick this one now.

.gitignore

@@ -1,6 +1,6 @@
 __pycache__
 GW150914
-GW150914-origin
+GW150914*
 docs
 *.tmp
-
+.codex

.idea/vcs.xml

@@ -1,6 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<project version="4">
-  <component name="VcsDirectoryMappings">
-    <mapping directory="" vcs="Git" />
-  </component>
-</project>

AMSS_NCKU_Input.py

@@ -16,9 +16,9 @@ import numpy
 File_directory   = "GW150914"                    ## output file directory
 Output_directory = "binary_output"               ## binary data file directory
                                                  ## The file directory name should not be too long
-MPI_processes    = 64                             ## number of mpi processes used in the simulation
+MPI_processes    = 2                             ## number of mpi processes used in the simulation
 
-GPU_Calculation  = "no"                          ## Use GPU or not 
+GPU_Calculation  = "yes"                          ## Use GPU or not 
                                                  ## (prefer "no" in the current version, because the GPU part may have bugs when integrated in this Python interface)
 CPU_Part         = 1.0
 GPU_Part         = 0.0
@@ -31,7 +31,7 @@ GPU_Part         = 0.0
 ## Setting the physical system and numerical method
 
 Symmetry                 = "equatorial-symmetry"   ## Symmetry of System: choose equatorial-symmetry、no-symmetry、octant-symmetry
-Equation_Class           = "BSSN"                  ## Evolution Equation: choose "BSSN", "BSSN-EScalar", "BSSN-EM", "Z4C" 
+Equation_Class           = "BSSN-EScalar"                  ## Evolution Equation: choose "BSSN", "BSSN-EScalar", "BSSN-EM", "Z4C"
                                                    ## If "BSSN-EScalar" is chosen, it is necessary to set other parameters below
 Initial_Data_Method      = "Ansorg-TwoPuncture"    ## initial data method: choose "Ansorg-TwoPuncture", "Lousto-Analytical", "Cao-Analytical", "KerrSchild-Analytical"
 Time_Evolution_Method    = "runge-kutta-45"        ## time evolution method: choose "runge-kutta-45"

AMSS_NCKU_Program.py

@@ -58,31 +58,36 @@ File_directory = os.path.join(input_data.File_directory)
 
 ## If the specified output directory exists, ask the user whether to continue
 if os.path.exists(File_directory):
-    print( " Output dictionary has been existed !!!  "                                                              )
+    auto_overwrite = str(getattr(input_data, "Auto_Overwrite_Output", "yes")).strip().lower()
-    print( " If you want to overwrite the existing file directory, please input 'continue' in the terminal !! "     ) 
+    if auto_overwrite in ("1", "yes", "y", "true", "on", "continue"):
-    print( " If you want to retain the existing file directory, please input 'stop' in the terminal to stop the "   ) 
+        print( " Output dictionary has been existed; Auto_Overwrite_Output=yes, continue the calculation. " )
-    print( " simulation. Then you can reset the output dictionary in the input script file AMSS_NCKU_Input.py !!! " )
+        print(                                                                                              )
-    print(                                                                                                          )
+    else:
-    ## Prompt whether to overwrite the existing directory
+        print( " Output dictionary has been existed !!!  "                                                              )
-    while True:
+        print( " If you want to overwrite the existing file directory, please input 'continue' in the terminal !! "     ) 
-        try:
+        print( " If you want to retain the existing file directory, please input 'stop' in the terminal to stop the "   ) 
-            inputvalue = input()
+        print( " simulation. Then you can reset the output dictionary in the input script file AMSS_NCKU_Input.py !!! " )
-            ## If the user agrees to overwrite, proceed and remove the existing directory
+        print(                                                                                                          )
-            if ( inputvalue == "continue" ):
+        ## Prompt whether to overwrite the existing directory
-                print( " Continue the calculation !!! " )
+        while True:
-                print(                                  )
+            try:
-                break  
+                inputvalue = input()
-            ## If the user chooses not to overwrite, exit and keep the existing directory
+                ## If the user agrees to overwrite, proceed and remove the existing directory
-            elif ( inputvalue == "stop" ):
+                if ( inputvalue == "continue" ):
-                print( " Stop the calculation !!! "    )
+                    print( " Continue the calculation !!! " )
-                sys.exit() 
+                    print(                                  )
-            ## If the user input is invalid, prompt again
+                    break  
-            else:
+                ## If the user chooses not to overwrite, exit and keep the existing directory
+                elif ( inputvalue == "stop" ):
+                    print( " Stop the calculation !!! "    )
+                    sys.exit() 
+                ## If the user input is invalid, prompt again
+                else:
+                    print( " Please input your choice !!! "                   )
+                    print( " Input 'continue' or 'stop' in the terminal !!! " )
+            except ValueError:
                 print( " Please input your choice !!! "                   )
                 print( " Input 'continue' or 'stop' in the terminal !!! " )
-        except ValueError:
-            print( " Please input your choice !!! "                   )
-            print( " Input 'continue' or 'stop' in the terminal !!! " )
         
 ## Remove the existing output directory if present
 shutil.rmtree(File_directory, ignore_errors=True)
@@ -126,6 +131,11 @@ setup.generate_AMSSNCKU_input()
 #inputvalue = input()  ## Wait for user input (press Enter) to proceed
 #print()
 
+setup.print_puncture_information()
+
+
+##################################################################
+
 ## Generate AMSS-NCKU program input files based on the configured parameters
 
 print(                                                                                   )
@@ -253,7 +263,7 @@ print()
 if (input_data.GPU_Calculation == "no"):
     ABE_file = os.path.join(AMSS_NCKU_source_copy, "ABE")
 elif (input_data.GPU_Calculation == "yes"):
-    ABE_file = os.path.join(AMSS_NCKU_source_copy, "ABEGPU")
+    ABE_file = os.path.join(AMSS_NCKU_source_copy, "ABE_CUDA")
 
 if not os.path.exists( ABE_file ):
     print(                                                                                                  )
@@ -307,7 +317,7 @@ if (input_data.Initial_Data_Method == "Ansorg-TwoPuncture" ):
     
     import generate_TwoPuncture_input
     
-    generate_TwoPuncture_input.generate_AMSSNCKU_TwoPuncture_input(numerical_grid.puncture_data)
+    generate_TwoPuncture_input.generate_AMSSNCKU_TwoPuncture_input()
     
     print(                                                                                              )
     print( " The input parfile for the TwoPunctureABE executable has been generated. " )
@@ -349,7 +359,7 @@ if (input_data.Initial_Data_Method == "Ansorg-TwoPuncture" ):
 
 import renew_puncture_parameter
     
-renew_puncture_parameter.append_AMSSNCKU_BSSN_input(File_directory, output_directory, numerical_grid.puncture_data)
+renew_puncture_parameter.append_AMSSNCKU_BSSN_input(File_directory, output_directory)
 
 
 ## Generated AMSS-NCKU input filename

AMSS_NCKU_source/BSSN_GPU/bssn_gpu.h

@@ -1,73 +0,0 @@
-
-#ifndef BSSN_GPU_H_
-#define BSSN_GPU_H_
-#include "bssn_macro.h"
-#include "macrodef.fh"
-
-#define DEVICE_ID 0
-// #define DEVICE_ID_BY_MPI_RANK
-#define GRID_DIM 256
-#define BLOCK_DIM 128
-
-#define _FH2_(i, j, k) fh[(i) + (j) * _1D_SIZE[2] + (k) * _2D_SIZE[2]]
-#define _FH3_(i, j, k) fh[(i) + (j) * _1D_SIZE[3] + (k) * _2D_SIZE[3]]
-#define pow2(x) ((x) * (x))
-#define TimeBetween(a, b) ((b.tv_sec - a.tv_sec) + (b.tv_usec - a.tv_usec) / 1000000.0f)
-#define M_ metac.
-#define Mh_ meta->
-#define Ms_ metassc.
-#define Msh_ metass->
-
-// #define TIMING
-
-#define RHS_SS_PARA int calledby, int mpi_rank, int *ex, double &T, double *crho, double *sigma, double *R, double *X, double *Y, double *Z, double *drhodx, double *drhody, double *drhodz, double *dsigmadx, double *dsigmady, double *dsigmadz, double *dRdx, double *dRdy, double *dRdz, double *drhodxx, double *drhodxy, double *drhodxz, double *drhodyy, double *drhodyz, double *drhodzz, double *dsigmadxx, double *dsigmadxy, double *dsigmadxz, double *dsigmadyy, double *dsigmadyz, double *dsigmadzz, double *dRdxx, double *dRdxy, double *dRdxz, double *dRdyy, double *dRdyz, double *dRdzz, double *chi, double *trK, double *dxx, double *gxy, double *gxz, double *dyy, double *gyz, double *dzz, double *Axx, double *Axy, double *Axz, double *Ayy, double *Ayz, double *Azz, double *Gamx, double *Gamy, double *Gamz, double *Lap, double *betax, double *betay, double *betaz, double *dtSfx, double *dtSfy, double *dtSfz, double *chi_rhs, double *trK_rhs, double *gxx_rhs, double *gxy_rhs, double *gxz_rhs, double *gyy_rhs, double *gyz_rhs, double *gzz_rhs, double *Axx_rhs, double *Axy_rhs, double *Axz_rhs, double *Ayy_rhs, double *Ayz_rhs, double *Azz_rhs, double *Gamx_rhs, double *Gamy_rhs, double *Gamz_rhs, double *Lap_rhs, double *betax_rhs, double *betay_rhs, double *betaz_rhs, double *dtSfx_rhs, double *dtSfy_rhs, double *dtSfz_rhs, double *rho, double *Sx, double *Sy, double *Sz, double *Sxx, double *Sxy, double *Sxz, double *Syy, double *Syz, double *Szz, double *Gamxxx, double *Gamxxy, double *Gamxxz, double *Gamxyy, double *Gamxyz, double *Gamxzz, double *Gamyxx, double *Gamyxy, double *Gamyxz, double *Gamyyy, double *Gamyyz, double *Gamyzz, double *Gamzxx, double *Gamzxy, double *Gamzxz, double *Gamzyy, double *Gamzyz, double *Gamzzz, double *Rxx, double *Rxy, double *Rxz, double *Ryy, double *Ryz, double *Rzz, double *ham_Res, double *movx_Res, double *movy_Res, double *movz_Res, double *Gmx_Res, double *Gmy_Res, double *Gmz_Res, int &Symmetry, int &Lev, double &eps, int &sst, int &co
-
-/**  main function */
-int gpu_rhs(int calledby, int mpi_rank, int *ex, double &T,
-            double *X, double *Y, double *Z,
-
-            double *chi, double *trK,
-
-            double *dxx, double *gxy, double *gxz, double *dyy, double *gyz, double *dzz,
-
-            double *Axx, double *Axy, double *Axz, double *Ayy, double *Ayz, double *Azz,
-
-            double *Gamx, double *Gamy, double *Gamz,
-
-            double *Lap, double *betax, double *betay, double *betaz,
-
-            double *dtSfx, double *dtSfy, double *dtSfz,
-
-            double *chi_rhs, double *trK_rhs,
-
-            double *gxx_rhs, double *gxy_rhs, double *gxz_rhs, double *gyy_rhs, double *gyz_rhs, double *gzz_rhs,
-
-            double *Axx_rhs, double *Axy_rhs, double *Axz_rhs, double *Ayy_rhs, double *Ayz_rhs, double *Azz_rhs,
-
-            double *Gamx_rhs, double *Gamy_rhs, double *Gamz_rhs,
-
-            double *Lap_rhs, double *betax_rhs, double *betay_rhs, double *betaz_rhs,
-
-            double *dtSfx_rhs, double *dtSfy_rhs, double *dtSfz_rhs,
-
-            double *rho, double *Sx, double *Sy, double *Sz, double *Sxx,
-            double *Sxy, double *Sxz, double *Syy, double *Syz, double *Szz,
-
-            double *Gamxxx, double *Gamxxy, double *Gamxxz, double *Gamxyy, double *Gamxyz, double *Gamxzz,
-
-            double *Gamyxx, double *Gamyxy, double *Gamyxz, double *Gamyyy, double *Gamyyz, double *Gamyzz,
-
-            double *Gamzxx, double *Gamzxy, double *Gamzxz, double *Gamzyy, double *Gamzyz, double *Gamzzz,
-
-            double *Rxx, double *Rxy, double *Rxz, double *Ryy, double *Ryz, double *Rzz,
-
-            double *ham_Res, double *movx_Res, double *movy_Res, double *movz_Res,
-            double *Gmx_Res, double *Gmy_Res, double *Gmz_Res,
-            int &Symmetry, int &Lev, double &eps, int &co);
-
-int gpu_rhs_ss(RHS_SS_PARA);
-
-/** Init GPU side data in GPUMeta. */
-// void init_fluid_meta_gpu(GPUMeta *gpu_meta);
-
-#endif

AMSS_NCKU_source/BSSN_GPU/bssn_gpu_class.h

@@ -1,210 +0,0 @@
-
-#ifndef BSSN_GPU_CLASS_H
-#define BSSN_GPU_CLASS_H
-
-#ifdef newc
-#include <iostream>
-#include <iomanip>
-#include <fstream>
-#include <cstdlib>
-#include <string>
-#include <cmath>
-using namespace std;
-#else
-#include <iostream.h>
-#include <iomanip.h>
-#include <fstream.h>
-#include <stdlib.h>
-#include <string.h>
-#include <math.h>
-#endif
-
-#include <mpi.h>
-
-#include "macrodef.h"
-#include "cgh.h"
-#include "ShellPatch.h"
-#include "misc.h"
-#include "var.h"
-#include "MyList.h"
-#include "monitor.h"
-#include "surface_integral.h"
-#include "checkpoint.h"
-
-// added by yangquan
-#include "bssn_macro.h"
-
-extern void setpbh(int iBHN, double **iPBH, double *iMass, int rBHN);
-
-class bssn_class
-{
-public:
-       // added by yangquan
-       //----------------------
-       int gpu_num_mynode;
-       int cpu_core_num_mynode;
-       int mpi_process_num_mynode;
-       int my_sequence_mynode;
-       int mynode_id;
-       int use_gpu;
-
-       virtual void Step_GPU(int lev, int YN);
-       virtual void Get_runtime_envirment();
-       // virtual void Step_OPENMP(int lev,int YN);
-       //----------------------
-
-       int ngfs;
-       int nprocs, myrank;
-       cgh *GH;
-       ShellPatch *SH;
-       double PhysTime;
-
-       int checkrun;
-       char checkfilename[50];
-       int Steps;
-       double StartTime, TotalTime;
-       double AnasTime, DumpTime, d2DumpTime, CheckTime;
-       double LastAnas, LastConsOut;
-       double Courant;
-       double numepss, numepsb, numepsh;
-       int Symmetry;
-       int maxl, decn;
-       double maxrex, drex;
-       int trfls, a_lev;
-
-       double dT;
-       double chitiny;
-
-       double **Porg0, **Porgbr, **Porg, **Porg1, **Porg_rhs;
-       int BH_num, BH_num_input;
-       double *Mass, *Pmom, *Spin;
-       double ADMMass;
-
-       var *phio, *trKo;
-       var *gxxo, *gxyo, *gxzo, *gyyo, *gyzo, *gzzo;
-       var *Axxo, *Axyo, *Axzo, *Ayyo, *Ayzo, *Azzo;
-       var *Gmxo, *Gmyo, *Gmzo;
-       var *Lapo, *Sfxo, *Sfyo, *Sfzo;
-       var *dtSfxo, *dtSfyo, *dtSfzo;
-
-       var *phi0, *trK0;
-       var *gxx0, *gxy0, *gxz0, *gyy0, *gyz0, *gzz0;
-       var *Axx0, *Axy0, *Axz0, *Ayy0, *Ayz0, *Azz0;
-       var *Gmx0, *Gmy0, *Gmz0;
-       var *Lap0, *Sfx0, *Sfy0, *Sfz0;
-       var *dtSfx0, *dtSfy0, *dtSfz0;
-
-       var *phi, *trK;
-       var *gxx, *gxy, *gxz, *gyy, *gyz, *gzz;
-       var *Axx, *Axy, *Axz, *Ayy, *Ayz, *Azz;
-       var *Gmx, *Gmy, *Gmz;
-       var *Lap, *Sfx, *Sfy, *Sfz;
-       var *dtSfx, *dtSfy, *dtSfz;
-
-       var *phi1, *trK1;
-       var *gxx1, *gxy1, *gxz1, *gyy1, *gyz1, *gzz1;
-       var *Axx1, *Axy1, *Axz1, *Ayy1, *Ayz1, *Azz1;
-       var *Gmx1, *Gmy1, *Gmz1;
-       var *Lap1, *Sfx1, *Sfy1, *Sfz1;
-       var *dtSfx1, *dtSfy1, *dtSfz1;
-
-       var *phi_rhs, *trK_rhs;
-       var *gxx_rhs, *gxy_rhs, *gxz_rhs, *gyy_rhs, *gyz_rhs, *gzz_rhs;
-       var *Axx_rhs, *Axy_rhs, *Axz_rhs, *Ayy_rhs, *Ayz_rhs, *Azz_rhs;
-       var *Gmx_rhs, *Gmy_rhs, *Gmz_rhs;
-       var *Lap_rhs, *Sfx_rhs, *Sfy_rhs, *Sfz_rhs;
-       var *dtSfx_rhs, *dtSfy_rhs, *dtSfz_rhs;
-
-       var *rho, *Sx, *Sy, *Sz, *Sxx, *Sxy, *Sxz, *Syy, *Syz, *Szz;
-
-       var *Gamxxx, *Gamxxy, *Gamxxz, *Gamxyy, *Gamxyz, *Gamxzz;
-       var *Gamyxx, *Gamyxy, *Gamyxz, *Gamyyy, *Gamyyz, *Gamyzz;
-       var *Gamzxx, *Gamzxy, *Gamzxz, *Gamzyy, *Gamzyz, *Gamzzz;
-
-       var *Rxx, *Rxy, *Rxz, *Ryy, *Ryz, *Rzz;
-
-       var *Rpsi4, *Ipsi4;
-       var *t1Rpsi4, *t1Ipsi4, *t2Rpsi4, *t2Ipsi4;
-
-       var *Cons_Ham, *Cons_Px, *Cons_Py, *Cons_Pz, *Cons_Gx, *Cons_Gy, *Cons_Gz;
-
-#ifdef Point_Psi4
-       var *phix, *phiy, *phiz;
-       var *trKx, *trKy, *trKz;
-       var *Axxx, *Axxy, *Axxz;
-       var *Axyx, *Axyy, *Axyz;
-       var *Axzx, *Axzy, *Axzz;
-       var *Ayyx, *Ayyy, *Ayyz;
-       var *Ayzx, *Ayzy, *Ayzz;
-       var *Azzx, *Azzy, *Azzz;
-#endif
-       // FIXME: uc = StateList, up = OldStateList, upp = SynchList_cor; so never touch these three data
-       MyList<var> *StateList, *SynchList_pre, *SynchList_cor, *RHSList;
-       MyList<var> *OldStateList, *DumpList;
-       MyList<var> *ConstraintList;
-
-       monitor *ErrorMonitor, *Psi4Monitor, *BHMonitor, *MAPMonitor;
-       monitor *ConVMonitor;
-       surface_integral *Waveshell;
-       checkpoint *CheckPoint;
-
-public:
-       bssn_class(double Couranti, double StartTimei, double TotalTimei, double DumpTimei, double d2DumpTimei, double CheckTimei, double AnasTimei,
-                  int Symmetryi, int checkruni, char *checkfilenamei, double numepssi, double numepsbi, double numepshi,
-                  int a_levi, int maxli, int decni, double maxrexi, double drexi);
-       ~bssn_class();
-
-       void Evolve(int Steps);
-       void RecursiveStep(int lev);
-#if (PSTR == 1)
-       void ParallelStep();
-       void SHStep();
-#endif
-       void RestrictProlong(int lev, int YN, bool BB, MyList<var> *SL, MyList<var> *OL, MyList<var> *corL);
-       void RestrictProlong_aux(int lev, int YN, bool BB, MyList<var> *SL, MyList<var> *OL, MyList<var> *corL);
-       void RestrictProlong(int lev, int YN, bool BB);
-       void ProlongRestrict(int lev, int YN, bool BB);
-       void Setup_Black_Hole_position();
-       void compute_Porg_rhs(double **BH_PS, double **BH_RHS, var *forx, var *fory, var *forz, int lev);
-       bool read_Pablo_file(int *ext, double *datain, char *filename);
-       void write_Pablo_file(int *ext, double xmin, double xmax, double ymin, double ymax, double zmin, double zmax,
-                             char *filename);
-       void AnalysisStuff(int lev, double dT_lev);
-       void Setup_KerrSchild();
-       void Enforce_algcon(int lev, int fg);
-
-       void testRestrict();
-       void testOutBd();
-
-       virtual void Setup_Initial_Data_Lousto();
-       virtual void Setup_Initial_Data_Cao();
-       virtual void Initialize();
-       virtual void Read_Ansorg();
-       virtual void Read_Pablo() {};
-       virtual void Compute_Psi4(int lev);
-       virtual void Step(int lev, int YN);
-       virtual void Interp_Constraint(bool infg);
-       virtual void Constraint_Out();
-       virtual void Compute_Constraint();
-
-#ifdef With_AHF
-protected:
-       MyList<var> *AHList, *AHDList, *GaugeList;
-       int AHfindevery;
-       double AHdumptime;
-       int *lastahdumpid, HN_num; // number of possible horizons
-       int *findeveryl;
-       double *xc, *yc, *zc, *xr, *yr, *zr;
-       bool *trigger;
-       double *dTT;
-       int *dumpid;
-
-public:
-       void AH_Prepare_derivatives();
-       bool AH_Interp_Points(MyList<var> *VarList,
-                             int NN, double **XX,
-                             double *Shellf, int Symmetryi);
-       void AH_Step_Find(int lev, double dT_lev);
-#endif
-};
-#endif /* BSSN_GPU_CLASS_H */

AMSS_NCKU_source/Block.C

@@ -12,7 +12,61 @@ using namespace std;
 #include "Block.h"
 #include "misc.h"
 
-Block::Block(int DIM, int *shapei, double *bboxi, int ranki, int ingfsi, int fngfsi, int levi, const int cgpui) : rank(ranki), ingfs(ingfsi), fngfs(fngfsi), lev(levi), cgpu(cgpui)
+#if USE_CUDA_BSSN || USE_CUDA_Z4C
+#include <cuda_runtime_api.h>
+#endif
+
+namespace {
+
+bool cuda_pin_gridfuncs_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_CUDA_PIN_GRIDFUNCS");
+    enabled = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+double *alloc_gridfunc(size_t count, unsigned char &pinned)
+{
+  pinned = 0;
+#if USE_CUDA_BSSN || USE_CUDA_Z4C
+  if (cuda_pin_gridfuncs_enabled())
+  {
+    double *ptr = 0;
+    cudaError_t err = cudaMallocHost((void **)&ptr, count * sizeof(double));
+    if (err == cudaSuccess)
+    {
+      pinned = 1;
+      return ptr;
+    }
+    cudaGetLastError();
+  }
+#endif
+  return (double *)malloc(sizeof(double) * count);
+}
+
+void free_gridfunc(double *ptr, unsigned char pinned)
+{
+  if (!ptr)
+    return;
+#if USE_CUDA_BSSN || USE_CUDA_Z4C
+  if (pinned)
+  {
+    cudaFreeHost(ptr);
+    return;
+  }
+#else
+  (void)pinned;
+#endif
+  free(ptr);
+}
+
+}
+
+Block::Block(int DIM, int *shapei, double *bboxi, int ranki, int ingfsi, int fngfsi, int levi, const int cgpui) : rank(ranki), lev(levi), cgpu(cgpui), ingfs(ingfsi), fngfs(fngfsi), igfs(0), fgfs(0), fgfs_pinned(0)
 {
   for (int i = 0; i < dim; i++)
     X[i] = 0;
@@ -70,9 +124,10 @@ Block::Block(int DIM, int *shapei, double *bboxi, int ranki, int ingfsi, int fng
 
     int nn = shape[0] * shape[1] * shape[2];
     fgfs = new double *[fngfs];
+    fgfs_pinned = new unsigned char[fngfs];
     for (int i = 0; i < fngfs; i++)
     {
-      fgfs[i] = (double *)malloc(sizeof(double) * nn);
+      fgfs[i] = alloc_gridfunc((size_t)nn, fgfs_pinned[i]);
       if (!(fgfs[i]))
       {
         cout << "on node#" << rank << ", out of memory when constructing Block." << endl;
@@ -107,11 +162,13 @@ Block::~Block()
       free(igfs[i]);
     delete[] igfs;
     for (int i = 0; i < fngfs; i++)
-      free(fgfs[i]);
+      free_gridfunc(fgfs[i], fgfs_pinned ? fgfs_pinned[i] : 0);
     delete[] fgfs;
+    delete[] fgfs_pinned;
     X[0] = X[1] = X[2] = 0;
     igfs = 0;
     fgfs = 0;
+    fgfs_pinned = 0;
   }
 }
 void Block::checkBlock()
@@ -187,6 +244,8 @@ void Block::swapList(MyList<var> *VarList1, MyList<var> *VarList2, int myrank)
     while (varl1 && varl2)
     {
       misc::swap<double *>(fgfs[varl1->data->sgfn], fgfs[varl2->data->sgfn]);
+      if (fgfs_pinned)
+        misc::swap<unsigned char>(fgfs_pinned[varl1->data->sgfn], fgfs_pinned[varl2->data->sgfn]);
       varl1 = varl1->next;
       varl2 = varl2->next;
     }

AMSS_NCKU_source/Block.h

@@ -18,9 +18,10 @@ public:
    int ingfs, fngfs;
    int *(*igfs);
    double *(*fgfs);
+   unsigned char *fgfs_pinned;
 
 public:
-   Block() {};
+   Block() : rank(0), lev(0), cgpu(0), ingfs(0), fngfs(0), igfs(0), fgfs(0), fgfs_pinned(0) {};
    Block(int DIM, int *shapei, double *bboxi, int ranki, int ingfsi, int fngfs, int levi, const int cgpui = 0);
 
    ~Block();

AMSS_NCKU_source/MPatch.C

@@ -14,6 +14,9 @@ using namespace std;
 #include "MPatch.h"
 #include "Parallel.h"
 #include "fmisc.h"
+#if USE_CUDA_BSSN
+#include "bssn_rhs_cuda.h"
+#endif
 #ifdef INTERP_LB_PROFILE
 #include "interp_lb_profile.h"
 #endif
@@ -178,6 +181,444 @@ int find_block_index_for_point(const BlockBinIndex &index, const double *pox, co
 
   return -1;
 }
+
+inline int fortran_idint_local(double x)
+{
+  return int(x);
+}
+
+bool interp_fast_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_INTERP_FAST");
+    enabled = (!env || atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+bool interp_gpu_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_INTERP_GPU");
+    enabled = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+bool interp_fast_compare_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_INTERP_FAST_COMPARE");
+    enabled = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+double interp_fast_compare_tol()
+{
+  static double tol = -1.0;
+  if (tol < 0.0)
+  {
+    const char *env = getenv("AMSS_INTERP_FAST_COMPARE_TOL");
+    tol = (env && atof(env) > 0.0) ? atof(env) : 1.0e-11;
+  }
+  return tol;
+}
+
+long long interp_fast_compare_limit()
+{
+  static long long limit = -1;
+  if (limit < 0)
+  {
+    const char *env = getenv("AMSS_INTERP_FAST_COMPARE_LIMIT");
+    limit = (env && atoll(env) > 0) ? atoll(env) : 4096;
+  }
+  return limit;
+}
+
+struct FastInterpStencil
+{
+  int cxB[dim];
+  double cx[dim];
+  double wx[8];
+  double wy[8];
+  double wz[8];
+  int nsamples;
+  int loc[512];
+  unsigned char sign_mask[512];
+  double weight[512];
+};
+
+inline void lagrange_unit_weights(double x, int ordn, double *w)
+{
+  for (int i = 0; i < ordn; i++)
+  {
+    double num = 1.0;
+    double den = 1.0;
+    for (int j = 0; j < ordn; j++)
+    {
+      if (j == i)
+        continue;
+      num *= (x - double(j));
+      den *= double(i - j);
+    }
+    w[i] = num / den;
+  }
+}
+
+inline void z_unit_weights(double x, int ordn, double *w)
+{
+  if (ordn == 6)
+  {
+    static const double c_uniform[6] = {-1.0, 5.0, -10.0, 10.0, -5.0, 1.0};
+    for (int i = 0; i < 6; i++)
+    {
+      if (x == double(i))
+      {
+        for (int j = 0; j < 6; j++)
+          w[j] = (j == i) ? 1.0 : 0.0;
+        return;
+      }
+    }
+    double den = 0.0;
+    for (int i = 0; i < 6; i++)
+    {
+      w[i] = c_uniform[i] / (x - double(i));
+      den += w[i];
+    }
+    for (int i = 0; i < 6; i++)
+      w[i] /= den;
+    return;
+  }
+  lagrange_unit_weights(x, ordn, w);
+}
+
+inline bool fast_interp_map_index(int idx, int extent, int d,
+                                  int &mapped, unsigned char &mask)
+{
+  if (idx > 0)
+    mapped = idx;
+  else
+  {
+    mask |= (unsigned char)(1u << d);
+#ifdef Vertex
+#ifdef Cell
+#error Both Cell and Vertex are defined
+#endif
+    mapped = 2 - idx;
+#else
+#ifdef Cell
+    mapped = 1 - idx;
+#else
+#error Not define Vertex nor Cell
+#endif
+#endif
+  }
+  return mapped >= 1 && mapped <= extent;
+}
+
+bool prepare_fast_interp_stencil(Block *BP, const double *pox, int ordn,
+                                 int Symmetry, FastInterpStencil &st)
+{
+  if (!BP || ordn <= 0 || ordn > 8)
+    return false;
+
+  st.nsamples = 0;
+
+  const int NO_SYMM = 0;
+  const int OCTANT = 2;
+  int cmin[dim], cmax[dim], cxT[dim];
+  for (int d = 0; d < dim; d++)
+  {
+    const double *X = BP->X[d];
+    const double dX = X[1] - X[0];
+    const int cxI = fortran_idint_local((pox[d] - X[0]) / dX + 0.4) + 1;
+    st.cxB[d] = cxI - ordn / 2 + 1;
+    cxT[d] = st.cxB[d] + ordn - 1;
+    cmin[d] = 1;
+    cmax[d] = BP->shape[d];
+
+#ifdef Vertex
+#ifdef Cell
+#error Both Cell and Vertex are defined
+#endif
+    if (Symmetry == OCTANT && d < 2 && fabs(X[0]) < dX)
+      cmin[d] = -ordn / 2 + 2;
+    if (Symmetry != NO_SYMM && d == 2 && fabs(X[0]) < dX)
+      cmin[d] = -ordn / 2 + 2;
+#else
+#ifdef Cell
+    if (Symmetry == OCTANT && d < 2 && fabs(X[0]) < dX)
+      cmin[d] = -ordn / 2 + 1;
+    if (Symmetry != NO_SYMM && d == 2 && fabs(X[0]) < dX)
+      cmin[d] = -ordn / 2 + 1;
+#else
+#error Not define Vertex nor Cell
+#endif
+#endif
+
+    if (st.cxB[d] < cmin[d])
+    {
+      st.cxB[d] = cmin[d];
+      cxT[d] = st.cxB[d] + ordn - 1;
+    }
+    if (cxT[d] > cmax[d])
+    {
+      cxT[d] = cmax[d];
+      st.cxB[d] = cxT[d] + 1 - ordn;
+    }
+
+    if (st.cxB[d] > 0)
+      st.cx[d] = (pox[d] - X[st.cxB[d] - 1]) / dX;
+    else
+    {
+#ifdef Vertex
+#ifdef Cell
+#error Both Cell and Vertex are defined
+#endif
+      st.cx[d] = (pox[d] + X[1 - st.cxB[d]]) / dX;
+#else
+#ifdef Cell
+      st.cx[d] = (pox[d] + X[-st.cxB[d]]) / dX;
+#else
+#error Not define Vertex nor Cell
+#endif
+#endif
+    }
+  }
+
+  lagrange_unit_weights(st.cx[0], ordn, st.wx);
+  lagrange_unit_weights(st.cx[1], ordn, st.wy);
+  z_unit_weights(st.cx[2], ordn, st.wz);
+
+  for (int kk = 0; kk < ordn; kk++)
+  {
+    for (int jj = 0; jj < ordn; jj++)
+    {
+      for (int ii = 0; ii < ordn; ii++)
+      {
+        unsigned char mask = 0;
+        int ix, iy, iz;
+        if (!fast_interp_map_index(st.cxB[0] + ii, BP->shape[0], 0, ix, mask) ||
+            !fast_interp_map_index(st.cxB[1] + jj, BP->shape[1], 1, iy, mask) ||
+            !fast_interp_map_index(st.cxB[2] + kk, BP->shape[2], 2, iz, mask))
+          return false;
+        const int s = st.nsamples++;
+        st.loc[s] = (ix - 1) + (iy - 1) * BP->shape[0] +
+                    (iz - 1) * BP->shape[0] * BP->shape[1];
+        st.sign_mask[s] = mask;
+        st.weight[s] = st.wx[ii] * st.wy[jj] * st.wz[kk];
+      }
+    }
+  }
+  return true;
+}
+
+bool interpolate_var_list_with_stencil(Block *BP, MyList<var> *VarList,
+                                       int num_var, const double *pox,
+                                       int ordn, int Symmetry,
+                                       const FastInterpStencil &st,
+                                       double *out)
+{
+  if (num_var <= 0 || num_var > 128)
+    return false;
+
+  double *data_ptrs[128];
+  double *soa_ptrs[128];
+  var *vars[128];
+  MyList<var> *varl = VarList;
+  int k = 0;
+  while (varl)
+  {
+    if (k >= num_var)
+      return false;
+    vars[k] = varl->data;
+    data_ptrs[k] = BP->fgfs[vars[k]->sgfn];
+    soa_ptrs[k] = vars[k]->SoA;
+    out[k] = 0.0;
+    varl = varl->next;
+    k++;
+  }
+
+  if (k != num_var)
+    return false;
+
+  for (int s = 0; s < st.nsamples; s++)
+  {
+    const int loc = st.loc[s];
+    const double w = st.weight[s];
+    const unsigned char mask = st.sign_mask[s];
+    if (mask == 0)
+    {
+      for (int v = 0; v < num_var; v++)
+        out[v] += w * data_ptrs[v][loc];
+    }
+    else
+    {
+      for (int v = 0; v < num_var; v++)
+      {
+        const double *SoA = soa_ptrs[v];
+        double sgn = 1.0;
+        if (mask & 1u)
+          sgn *= SoA[0];
+        if (mask & 2u)
+          sgn *= SoA[1];
+        if (mask & 4u)
+          sgn *= SoA[2];
+        out[v] += w * sgn * data_ptrs[v][loc];
+      }
+    }
+  }
+
+  if (interp_fast_compare_enabled())
+  {
+    static int report_count = 0;
+    static long long compare_calls = 0;
+    if (compare_calls++ >= interp_fast_compare_limit())
+      return true;
+    const double tol = interp_fast_compare_tol();
+    varl = VarList;
+    k = 0;
+    while (varl)
+    {
+      var *vp = vars[k];
+      double ref = 0.0;
+      double x = pox[0], y = pox[1], z = pox[2];
+      f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2],
+                      BP->fgfs[vp->sgfn], ref,
+                      x, y, z, ordn, vp->SoA, Symmetry);
+      const double diff = fabs(ref - out[k]);
+      const double scale = 1.0 + fabs(ref);
+      if (diff > tol * scale && report_count < 32)
+      {
+        int rank = 0;
+        MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+        fprintf(stderr,
+                "[AMSS-INTERP-CMP][rank %d] var=%s diff=%.17e ref=%.17e fast=%.17e p=(%.17e,%.17e,%.17e)\n",
+                rank, vp->name, diff, ref, out[k], pox[0], pox[1], pox[2]);
+        report_count++;
+      }
+      varl = varl->next;
+      k++;
+    }
+  }
+
+  return true;
+}
+
+bool interpolate_var_list_fast(Block *BP, MyList<var> *VarList, int num_var,
+                               const double *pox, int ordn, int Symmetry,
+                               double *out)
+{
+  if (!interp_fast_enabled())
+    return false;
+
+  FastInterpStencil st;
+  if (!prepare_fast_interp_stencil(BP, pox, ordn, Symmetry, st))
+    return false;
+
+  return interpolate_var_list_with_stencil(BP, VarList, num_var, pox,
+                                           ordn, Symmetry, st, out);
+}
+
+struct CachedInterpPoint
+{
+  Block *bp;
+  int owner_rank;
+  FastInterpStencil stencil;
+};
+
+struct SurfaceInterpCache
+{
+  Patch *patch;
+  int NN;
+  int symmetry;
+  double key[9];
+  vector<CachedInterpPoint> points;
+
+  SurfaceInterpCache() : patch(0), NN(0), symmetry(-1) {}
+};
+
+bool surface_cache_key_matches(const SurfaceInterpCache &cache, Patch *patch,
+                               int NN, double **XX, int Symmetry)
+{
+  if (cache.patch != patch || cache.NN != NN || cache.symmetry != Symmetry ||
+      int(cache.points.size()) != NN || NN <= 0)
+    return false;
+  const int mid = NN / 2;
+  const int last = NN - 1;
+  const int ids[3] = {0, mid, last};
+  int p = 0;
+  for (int q = 0; q < 3; q++)
+    for (int d = 0; d < dim; d++)
+      if (cache.key[p++] != XX[d][ids[q]])
+        return false;
+  return true;
+}
+
+SurfaceInterpCache *find_surface_cache(Patch *patch, int NN, double **XX,
+                                       int Symmetry)
+{
+  static vector<SurfaceInterpCache> caches;
+  for (size_t i = 0; i < caches.size(); i++)
+    if (surface_cache_key_matches(caches[i], patch, NN, XX, Symmetry))
+      return &caches[i];
+  if (caches.size() >= 24)
+    caches.erase(caches.begin());
+  caches.push_back(SurfaceInterpCache());
+  return &caches.back();
+}
+
+bool build_surface_cache(SurfaceInterpCache &cache, Patch *patch, int NN,
+                         double **XX, int Symmetry, const double *DH,
+                         const BlockBinIndex &block_index, int ordn)
+{
+  int myrank = 0;
+  MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
+  cache.patch = patch;
+  cache.NN = NN;
+  cache.symmetry = Symmetry;
+  cache.points.clear();
+  cache.points.resize(NN);
+  const int mid = NN / 2;
+  const int last = NN - 1;
+  const int ids[3] = {0, mid, last};
+  int p = 0;
+  for (int q = 0; q < 3; q++)
+    for (int d = 0; d < dim; d++)
+      cache.key[p++] = XX[d][ids[q]];
+
+  for (int j = 0; j < NN; j++)
+  {
+    double pox[dim];
+    for (int d = 0; d < dim; d++)
+      pox[d] = XX[d][j];
+    const int block_i = find_block_index_for_point(block_index, pox, DH);
+    if (block_i < 0)
+    {
+      cache.points[j].bp = 0;
+      cache.points[j].owner_rank = -1;
+      continue;
+    }
+    Block *BP = block_index.views[block_i].bp;
+    cache.points[j].bp = BP;
+    cache.points[j].owner_rank = BP->rank;
+    cache.points[j].stencil.nsamples = 0;
+    if (BP->rank == myrank)
+    {
+      if (!prepare_fast_interp_stencil(BP, pox, ordn, Symmetry,
+                                       cache.points[j].stencil))
+        return false;
+    }
+  }
+  return true;
+}
 } // namespace
 
 Patch::Patch(int DIM, int *shapei, double *bboxi, int levi, bool buflog, int Symmetry) : lev(levi)
@@ -565,14 +1006,18 @@ void Patch::Interp_Points(MyList<var> *VarList,
       if (myrank == BP->rank)
       {
         //---> interpolation
-        varl = VarList;
+        if (!interpolate_var_list_fast(BP, VarList, num_var, pox, ordn,
-        int k = 0;
+                                       Symmetry, Shellf + j * num_var))
-        while (varl) // run along variables
         {
-          f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl->data->sgfn], Shellf[j * num_var + k],
+          varl = VarList;
-                          pox[0], pox[1], pox[2], ordn, varl->data->SoA, Symmetry);
+          int k = 0;
-          varl = varl->next;
+          while (varl) // run along variables
-          k++;
+          {
+            f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl->data->sgfn], Shellf[j * num_var + k],
+                            pox[0], pox[1], pox[2], ordn, varl->data->SoA, Symmetry);
+            varl = varl->next;
+            k++;
+          }
         }
       }
     }
@@ -659,8 +1104,6 @@ void Patch::Interp_Points(MyList<var> *VarList,
     varl = varl->next;
   }
 
-  memset(Shellf, 0, sizeof(double) * NN * num_var);
-
   // owner_rank[j] records which MPI rank owns point j
   int *owner_rank;
   owner_rank = new int[NN];
@@ -672,8 +1115,113 @@ void Patch::Interp_Points(MyList<var> *VarList,
     DH[i] = getdX(i);
   BlockBinIndex block_index;
   build_block_bin_index(this, DH, block_index);
+  SurfaceInterpCache *surface_cache = 0;
+  bool use_surface_cache = false;
+  if (interp_fast_enabled())
+  {
+    surface_cache = find_surface_cache(this, NN, XX, Symmetry);
+    use_surface_cache = surface_cache_key_matches(*surface_cache, this, NN, XX, Symmetry);
+    if (!use_surface_cache)
+      use_surface_cache = build_surface_cache(*surface_cache, this, NN, XX,
+                                              Symmetry, DH, block_index, ordn);
+  }
 
   // --- Interpolation phase (identical to original) ---
+#if USE_CUDA_BSSN
+  const bool use_gpu_interp = interp_gpu_enabled() && use_surface_cache && num_var == 2 &&
+                              VarList && VarList->next && !VarList->next->next;
+#else
+  const bool use_gpu_interp = false;
+#endif
+  if (use_gpu_interp)
+  {
+#if USE_CUDA_BSSN
+    vector<vector<int> > local_points(block_index.views.size());
+    for (int j = 0; j < NN; j++)
+    {
+      for (int i = 0; i < dim; i++)
+      {
+        if (myrank == 0 && (XX[i][j] < bbox[i] + lli[i] * DH[i] || XX[i][j] > bbox[dim + i] - uui[i] * DH[i]))
+        {
+          cout << "Patch::Interp_Points: point (";
+          for (int k = 0; k < dim; k++)
+          {
+            cout << XX[k][j];
+            if (k < dim - 1)
+              cout << ",";
+            else
+              cout << ") is out of current Patch." << endl;
+          }
+          MPI_Abort(MPI_COMM_WORLD, 1);
+        }
+      }
+
+      CachedInterpPoint &cp = surface_cache->points[j];
+      Block *BP = cp.bp;
+      owner_rank[j] = cp.owner_rank;
+      if (BP && myrank == BP->rank)
+      {
+        for (size_t bi = 0; bi < block_index.views.size(); bi++)
+        {
+          if (block_index.views[bi].bp == BP)
+          {
+            local_points[bi].push_back(j);
+            break;
+          }
+        }
+      }
+    }
+
+    var *v0 = VarList->data;
+    var *v1 = VarList->next->data;
+    double soa6[6] = {
+        v0->SoA[0], v0->SoA[1], v0->SoA[2],
+        v1->SoA[0], v1->SoA[1], v1->SoA[2]};
+
+    for (size_t bi = 0; bi < local_points.size(); bi++)
+    {
+      const int count = int(local_points[bi].size());
+      if (count <= 0)
+        continue;
+
+      Block *BP = block_index.views[bi].bp;
+      vector<double> px(count), py(count), pz(count), out(2 * count);
+      for (int q = 0; q < count; q++)
+      {
+        const int j = local_points[bi][q];
+        px[q] = XX[0][j];
+        py[q] = XX[1][j];
+        pz[q] = XX[2][j];
+      }
+
+      const double dx = BP->X[0][1] - BP->X[0][0];
+      const double dy = BP->X[1][1] - BP->X[1][0];
+      const double dz = BP->X[2][1] - BP->X[2][0];
+      const int ok = bssn_cuda_interp_host_two_fields(
+          BP, BP->shape,
+          BP->fgfs[v0->sgfn], BP->fgfs[v1->sgfn],
+          BP->X[0][0], BP->X[1][0], BP->X[2][0],
+          dx, dy, dz,
+          &px[0], &py[0], &pz[0], count,
+          ordn, Symmetry, soa6, &out[0]);
+      if (ok != 0)
+      {
+        if (myrank == 0)
+          cout << "Patch::Interp_Points: CUDA two-field interpolation failed" << endl;
+        MPI_Abort(MPI_COMM_WORLD, 1);
+      }
+
+      for (int q = 0; q < count; q++)
+      {
+        const int j = local_points[bi][q];
+        Shellf[j * num_var] = out[2 * q];
+        Shellf[j * num_var + 1] = out[2 * q + 1];
+      }
+    }
+#endif
+  }
+  else
+  {
   for (int j = 0; j < NN; j++)
   {
     double pox[dim];
@@ -695,24 +1243,55 @@ void Patch::Interp_Points(MyList<var> *VarList,
       }
     }
 
-    const int block_i = find_block_index_for_point(block_index, pox, DH);
+    if (use_surface_cache)
-    if (block_i >= 0)
     {
-      Block *BP = block_index.views[block_i].bp;
+      CachedInterpPoint &cp = surface_cache->points[j];
-      owner_rank[j] = BP->rank;
+      Block *BP = cp.bp;
-      if (myrank == BP->rank)
+      owner_rank[j] = cp.owner_rank;
+      if (BP && myrank == BP->rank)
       {
-        varl = VarList;
+        if (!interpolate_var_list_with_stencil(BP, VarList, num_var, pox,
-        int k = 0;
+                                               ordn, Symmetry, cp.stencil,
-        while (varl)
+                                               Shellf + j * num_var))
         {
-          f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl->data->sgfn], Shellf[j * num_var + k],
+          MyList<var> *varl_fallback = VarList;
-                          pox[0], pox[1], pox[2], ordn, varl->data->SoA, Symmetry);
+          int k = 0;
-          varl = varl->next;
+          while (varl_fallback)
-          k++;
+          {
+            f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl_fallback->data->sgfn], Shellf[j * num_var + k],
+                            pox[0], pox[1], pox[2], ordn, varl_fallback->data->SoA, Symmetry);
+            varl_fallback = varl_fallback->next;
+            k++;
+          }
         }
       }
     }
+    else
+    {
+      const int block_i = find_block_index_for_point(block_index, pox, DH);
+      if (block_i >= 0)
+      {
+        Block *BP = block_index.views[block_i].bp;
+        owner_rank[j] = BP->rank;
+      if (myrank == BP->rank)
+      {
+        if (!interpolate_var_list_fast(BP, VarList, num_var, pox, ordn,
+                                       Symmetry, Shellf + j * num_var))
+        {
+          MyList<var> *varl_fallback = VarList;
+          int k = 0;
+          while (varl_fallback)
+          {
+            f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl_fallback->data->sgfn], Shellf[j * num_var + k],
+                            pox[0], pox[1], pox[2], ordn, varl_fallback->data->SoA, Symmetry);
+            varl_fallback = varl_fallback->next;
+            k++;
+          }
+        }
+      }
+      }
+    }
+  }
   }
 
 #ifdef INTERP_LB_PROFILE
@@ -969,14 +1548,18 @@ void Patch::Interp_Points(MyList<var> *VarList,
       if (myrank == BP->rank)
       {
         //---> interpolation
-        varl = VarList;
+        if (!interpolate_var_list_fast(BP, VarList, num_var, pox, ordn,
-        int k = 0;
+                                       Symmetry, Shellf + j * num_var))
-        while (varl) // run along variables
         {
-          f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl->data->sgfn], Shellf[j * num_var + k],
+          varl = VarList;
-                          pox[0], pox[1], pox[2], ordn, varl->data->SoA, Symmetry);
+          int k = 0;
-          varl = varl->next;
+          while (varl) // run along variables
-          k++;
+          {
+            f_global_interp(BP->shape, BP->X[0], BP->X[1], BP->X[2], BP->fgfs[varl->data->sgfn], Shellf[j * num_var + k],
+                            pox[0], pox[1], pox[2], ordn, varl->data->SoA, Symmetry);
+            varl = varl->next;
+            k++;
+          }
         }
       }
     }

AMSS_NCKU_source/Parallel.h

@@ -104,6 +104,14 @@ namespace Parallel
     double **recv_bufs;
     int *send_buf_caps;
     int *recv_buf_caps;
+    unsigned char *send_buf_pinned;
+    unsigned char *recv_buf_pinned;
+    unsigned char *send_buf_is_dev;
+    unsigned char *recv_buf_is_dev;
+    int *send_buf_caps_dev;
+    int *recv_buf_caps_dev;
+    double **send_bufs_dev;
+    double **recv_bufs_dev;
     MPI_Request *reqs;
     MPI_Status *stats;
     int max_reqs;
@@ -111,12 +119,14 @@ namespace Parallel
     int *tc_req_node;
     int *tc_req_is_recv;
     int *tc_completed;
+    bool cuda_aware_mode;
     SyncCache();
     void invalidate();
     void destroy();
   };
 
   void Sync_cached(MyList<Patch> *PatL, MyList<var> *VarList, int Symmetry, SyncCache &cache);
+  void Sync_ensure_cache(MyList<Patch> *PatL, int Symmetry, SyncCache &cache);
   void transfer_cached(MyList<gridseg> **src, MyList<gridseg> **dst,
                        MyList<var> *VarList1, MyList<var> *VarList2,
                        int Symmetry, SyncCache &cache);

AMSS_NCKU_source/Z4c_class.C

@@ -2,6 +2,7 @@
 #ifdef newc
 #include <sstream>
 #include <cstdio>
+#include <cstdlib>
 #include <map>
 using namespace std;
 #else
@@ -28,6 +29,14 @@ using namespace std;
 #include "kodiss.h"
 #include "parameters.h"
 
+#ifndef USE_CUDA_Z4C
+#define USE_CUDA_Z4C 0
+#endif
+
+#if USE_CUDA_Z4C && (ABEtype == 2)
+#include "z4c_rhs_cuda.h"
+#endif
+
 #ifdef With_AHF
 #include "derivatives.h"
 #include "myglobal.h"
@@ -118,6 +127,8 @@ void Z4c_class::Initialize()
     CheckPoint->readcheck_sh(SH, myrank);
 #endif
 
+  Initialize_Level_Runtime();
+
   double h = GH->PatL[0]->data->blb->data->getdX(0);
   for (int i = 1; i < dim; i++)
     h = Mymin(h, GH->PatL[0]->data->blb->data->getdX(i));
@@ -170,8 +181,749 @@ Z4c_class::~Z4c_class()
 #ifndef CPBC
 // for sommerfeld boundary
 
+#if USE_CUDA_Z4C && (ABEtype == 2)
+#ifdef WithShell
+#error "USE_CUDA_Z4C resident path currently supports Cartesian non-shell Z4C only"
+#endif
+#if (MRBD == 2)
+#error "USE_CUDA_Z4C resident path does not support MRBD == 2"
+#endif
+
+namespace {
+
+static const int k_z4c_cuda_bh_state_indices[3] = {18, 19, 20};
+
+bool fill_z4c_cuda_views(Block *cg, MyList<var> *vars,
+                         double **host_views,
+                         double *propspeeds = 0,
+                         double *soa_flat = 0)
+{
+  int idx = 0;
+  while (vars && idx < Z4C_CUDA_STATE_COUNT)
+  {
+    host_views[idx] = cg->fgfs[vars->data->sgfn];
+    if (propspeeds)
+      propspeeds[idx] = vars->data->propspeed;
+    if (soa_flat)
+    {
+      soa_flat[3 * idx + 0] = vars->data->SoA[0];
+      soa_flat[3 * idx + 1] = vars->data->SoA[1];
+      soa_flat[3 * idx + 2] = vars->data->SoA[2];
+    }
+    vars = vars->next;
+    ++idx;
+  }
+  return idx == Z4C_CUDA_STATE_COUNT && vars == 0;
+}
+
+bool z4c_cuda_keep_resident_after_step(int lev, int trfls_in, int analysis_lev)
+{
+  static int keep_all_levels = -1;
+  if (keep_all_levels < 0)
+  {
+    const char *env = getenv("AMSS_CUDA_KEEP_ALL_LEVELS");
+    keep_all_levels = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_CUDA_Z4C_KEEP_RESIDENT_AFTER_STEP");
+    if (env)
+      enabled = (atoi(env) != 0) ? 1 : 0;
+    else
+    {
+      env = getenv("AMSS_CUDA_KEEP_RESIDENT_AFTER_STEP");
+      enabled = (env && atoi(env) != 0) ? 1 : 0;
+    }
+  }
+  if (!enabled)
+    return false;
+  if (lev == analysis_lev)
+    return false;
+  if (keep_all_levels)
+    return true;
+  return lev < trfls_in;
+}
+
+void z4c_cuda_download_level_state(MyList<Patch> *PatL, MyList<var> *vars, int myrank, bool release_ctx)
+{
+  MyList<Patch> *Pp = PatL;
+  while (Pp)
+  {
+    MyList<Block> *BP = Pp->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank && z4c_cuda_has_resident_state(cg))
+      {
+        double *state_out[Z4C_CUDA_STATE_COUNT];
+        if (!fill_z4c_cuda_views(cg, vars, state_out))
+        {
+          cout << "CUDA Z4C state list mismatch on resident state download" << endl;
+          MPI_Abort(MPI_COMM_WORLD, 1);
+        }
+        if (z4c_cuda_download_resident_state(cg, cg->shape, state_out))
+        {
+          cout << "CUDA Z4C resident state download failed" << endl;
+          MPI_Abort(MPI_COMM_WORLD, 1);
+        }
+        if (release_ctx)
+          z4c_cuda_release_step_ctx(cg);
+      }
+      if (BP == Pp->data->ble)
+        break;
+      BP = BP->next;
+    }
+    Pp = Pp->next;
+  }
+}
+
+bool z4c_cuda_patch_contains_point(Patch *patch, const double *point)
+{
+  if (!patch)
+    return false;
+  for (int d = 0; d < dim; d++)
+  {
+    const double h = patch->getdX(d);
+    const double lo = patch->bbox[d] + patch->lli[d] * h;
+    const double hi = patch->bbox[dim + d] - patch->uui[d] * h;
+    if (point[d] < lo || point[d] > hi)
+      return false;
+  }
+  return true;
+}
+
+bool z4c_cuda_point_in_block(Patch *patch, Block *block,
+                             const double *point, const double *DH)
+{
+  if (!patch || !block)
+    return false;
+  for (int d = 0; d < dim; d++)
+  {
+    double llb;
+    double uub;
+#ifdef Vertex
+#ifdef Cell
+#error Both Cell and Vertex are defined
+#endif
+    llb = (feq(block->bbox[d], patch->bbox[d], DH[d] / 2))
+              ? block->bbox[d] + patch->lli[d] * DH[d]
+              : block->bbox[d] + (ghost_width - 0.5) * DH[d];
+    uub = (feq(block->bbox[dim + d], patch->bbox[dim + d], DH[d] / 2))
+              ? block->bbox[dim + d] - patch->uui[d] * DH[d]
+              : block->bbox[dim + d] - (ghost_width - 0.5) * DH[d];
+#else
+#ifdef Cell
+    llb = (feq(block->bbox[d], patch->bbox[d], DH[d] / 2))
+              ? block->bbox[d] + patch->lli[d] * DH[d]
+              : block->bbox[d] + ghost_width * DH[d];
+    uub = (feq(block->bbox[dim + d], patch->bbox[dim + d], DH[d] / 2))
+              ? block->bbox[dim + d] - patch->uui[d] * DH[d]
+              : block->bbox[dim + d] - ghost_width * DH[d];
+#else
+#error Not define Vertex nor Cell
+#endif
+#endif
+    if (point[d] - llb < -DH[d] / 2 || point[d] - uub > DH[d] / 2)
+      return false;
+  }
+  return true;
+}
+
+int z4c_cuda_interp_tile_start(const double *coords, int n, double x, double dx, int ordn)
+{
+  if (!coords || n <= ordn)
+    return 0;
+  int cxi = int((x - coords[0]) / dx + 0.4) + 1;
+  int start = cxi - ordn / 2;
+  if (start < 0)
+    start = 0;
+  const int max_start = n - ordn;
+  if (start > max_start)
+    start = max_start;
+  return start;
+}
+
+bool z4c_cuda_interp_bh_point_resident(MyList<Patch> *PatL,
+                                       int myrank,
+                                       const double *point,
+                                       var *forx, var *fory, var *forz,
+                                       int Symmetry,
+                                       double *shellf)
+{
+  const int ordn = 2 * ghost_width;
+  int owner_rank = -1;
+
+  shellf[0] = shellf[1] = shellf[2] = 0.0;
+
+  MyList<Patch> *PL = PatL;
+  while (PL)
+  {
+    Patch *patch = PL->data;
+    if (!z4c_cuda_patch_contains_point(patch, point))
+    {
+      PL = PL->next;
+      continue;
+    }
+
+    double DH[dim];
+    for (int d = 0; d < dim; d++)
+      DH[d] = patch->getdX(d);
+
+    MyList<Block> *BP = patch->blb;
+    while (BP)
+    {
+      Block *block = BP->data;
+      if (z4c_cuda_point_in_block(patch, block, point, DH))
+      {
+        owner_rank = block->rank;
+        if (myrank == owner_rank)
+        {
+          int interp_ordn = ordn;
+          int interp_sym = Symmetry;
+          double x = point[0];
+          double y = point[1];
+          double z = point[2];
+
+          if (z4c_cuda_has_resident_state(block) &&
+              block->shape[0] >= ordn && block->shape[1] >= ordn && block->shape[2] >= ordn)
+          {
+            var *vars[3] = {forx, fory, forz};
+            static int use_device_bh_interp = -1;
+            if (use_device_bh_interp < 0)
+            {
+              const char *env = getenv("AMSS_CUDA_Z4C_BH_INTERP_DEVICE");
+              use_device_bh_interp = (env && atoi(env) != 0) ? 1 : 0;
+            }
+            bool used_device_interp = false;
+            if (use_device_bh_interp)
+            {
+              double soa3[9];
+              for (int f = 0; f < 3; f++)
+              {
+                soa3[3 * f + 0] = vars[f]->SoA[0];
+                soa3[3 * f + 1] = vars[f]->SoA[1];
+                soa3[3 * f + 2] = vars[f]->SoA[2];
+              }
+              used_device_interp =
+                  (z4c_cuda_interp_state_point3(block, block->shape,
+                                                k_z4c_cuda_bh_state_indices[0],
+                                                k_z4c_cuda_bh_state_indices[1],
+                                                k_z4c_cuda_bh_state_indices[2],
+                                                block->X[0][0], block->X[1][0], block->X[2][0],
+                                                DH[0], DH[1], DH[2],
+                                                x, y, z,
+                                                interp_ordn, interp_sym,
+                                                soa3, shellf) == 0);
+            }
+            if (!used_device_interp)
+            {
+              double *shift_views[3] = {
+                  block->fgfs[forx->sgfn],
+                  block->fgfs[fory->sgfn],
+                  block->fgfs[forz->sgfn]};
+              if (z4c_cuda_download_state_subset(block, block->shape, 3,
+                                                 k_z4c_cuda_bh_state_indices,
+                                                 shift_views) != 0)
+              {
+                cout << "CUDA Z4C BH shift download failed" << endl;
+                MPI_Abort(MPI_COMM_WORLD, 1);
+              }
+              f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                              block->fgfs[forx->sgfn], shellf[0],
+                              x, y, z, interp_ordn, forx->SoA, interp_sym);
+              f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                              block->fgfs[fory->sgfn], shellf[1],
+                              x, y, z, interp_ordn, fory->SoA, interp_sym);
+              f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                              block->fgfs[forz->sgfn], shellf[2],
+                              x, y, z, interp_ordn, forz->SoA, interp_sym);
+            }
+          }
+          else
+          {
+            f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                            block->fgfs[forx->sgfn], shellf[0],
+                            x, y, z, interp_ordn, forx->SoA, interp_sym);
+            f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                            block->fgfs[fory->sgfn], shellf[1],
+                            x, y, z, interp_ordn, fory->SoA, interp_sym);
+            f_global_interp(block->shape, block->X[0], block->X[1], block->X[2],
+                            block->fgfs[forz->sgfn], shellf[2],
+                            x, y, z, interp_ordn, forz->SoA, interp_sym);
+          }
+        }
+        break;
+      }
+      if (BP == patch->ble)
+        break;
+      BP = BP->next;
+    }
+
+    if (owner_rank >= 0)
+      break;
+    PL = PL->next;
+  }
+
+  if (owner_rank < 0)
+    return false;
+
+  MPI_Bcast(shellf, 3, MPI_DOUBLE, owner_rank, MPI_COMM_WORLD);
+  return true;
+}
+
+bool z4c_cuda_compute_porg_rhs_resident(cgh *GH,
+                                        int ilev,
+                                        int myrank,
+                                        int BH_num,
+                                        double **BH_PS,
+                                        double **BH_RHS,
+                                        var *forx, var *fory, var *forz,
+                                        int Symmetry)
+{
+  for (int n = 0; n < BH_num; n++)
+  {
+    double shellf[3] = {0.0, 0.0, 0.0};
+    int lev = ilev;
+    while (lev >= 0 &&
+           !z4c_cuda_interp_bh_point_resident(GH->PatL[lev], myrank, BH_PS[n],
+                                              forx, fory, forz, Symmetry, shellf))
+    {
+      --lev;
+    }
+    if (lev < 0)
+      return false;
+    BH_RHS[n][0] = -shellf[0];
+    BH_RHS[n][1] = -shellf[1];
+    BH_RHS[n][2] = -shellf[2];
+  }
+  return true;
+}
+
+bool z4c_cuda_download_bh_shift_level(MyList<Patch> *PatL,
+                                      int myrank,
+                                      var *forx, var *fory, var *forz)
+{
+  MyList<Patch> *Pp = PatL;
+  while (Pp)
+  {
+    MyList<Block> *BP = Pp->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank && z4c_cuda_has_resident_state(cg))
+      {
+        double *fields[3] = {
+            cg->fgfs[forx->sgfn],
+            cg->fgfs[fory->sgfn],
+            cg->fgfs[forz->sgfn]};
+        if (z4c_cuda_download_state_subset(cg, cg->shape, 3,
+                                           k_z4c_cuda_bh_state_indices,
+                                           fields))
+          return false;
+      }
+      if (BP == Pp->data->ble)
+        break;
+      BP = BP->next;
+    }
+    Pp = Pp->next;
+  }
+  return true;
+}
+
+bool z4c_cuda_refresh_constraint_level(MyList<Patch> *PatL,
+                                       int myrank,
+                                       var *Cons_Ham, var *Cons_Px,
+                                       var *Cons_Py, var *Cons_Pz,
+                                       var *Cons_Gx, var *Cons_Gy,
+                                       var *Cons_Gz, var *TZ0,
+                                       int Symmetry, int lev, double eps)
+{
+  bool all_resident = true;
+  const int tz_index = 24;
+  MyList<Patch> *Pp = PatL;
+  while (Pp)
+  {
+    MyList<Block> *BP = Pp->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank)
+      {
+        if (!z4c_cuda_has_resident_state(cg))
+        {
+          all_resident = false;
+        }
+        else
+        {
+          double *constraints[7] = {
+              cg->fgfs[Cons_Ham->sgfn], cg->fgfs[Cons_Px->sgfn],
+              cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
+              cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn],
+              cg->fgfs[Cons_Gz->sgfn]};
+          double *tz_out[1] = {cg->fgfs[TZ0->sgfn]};
+          int co = 0;
+          if (z4c_cuda_compute_constraints_resident(cg, cg->shape,
+                                                   cg->X[0], cg->X[1], cg->X[2],
+                                                   Symmetry, eps, co,
+                                                   constraints) ||
+              z4c_cuda_download_state_subset(cg, cg->shape, 1, &tz_index, tz_out))
+          {
+            cout << "CUDA Z4C resident constraint refresh failed" << endl;
+            MPI_Abort(MPI_COMM_WORLD, 1);
+          }
+        }
+      }
+      if (BP == Pp->data->ble)
+        break;
+      BP = BP->next;
+    }
+    Pp = Pp->next;
+  }
+  return all_resident;
+}
+
+long long &z4c_constraint_output_counter()
+{
+  static long long counter = 0;
+  return counter;
+}
+
+int z4c_constraint_output_every()
+{
+  static int every = -1;
+  if (every < 0)
+  {
+    const char *env = getenv("AMSS_CUDA_Z4C_CONSTRAINT_EVERY");
+    every = (env && atoi(env) > 0) ? atoi(env) : 1;
+  }
+  return every;
+}
+
+bool z4c_constraint_output_due_now()
+{
+  const int every = z4c_constraint_output_every();
+  return every <= 1 || (z4c_constraint_output_counter() % every) == 0;
+}
+
+void z4c_constraint_output_advance()
+{
+  z4c_constraint_output_counter()++;
+}
+
+} // namespace
+#endif
+
 void Z4c_class::Step(int lev, int YN)
 {
+#if USE_CUDA_Z4C && (ABEtype == 2)
+  double dT_lev = dT * pow(0.5, Mymax(lev, trfls));
+#ifdef With_AHF
+  AH_Step_Find(lev, dT_lev);
+#endif
+  bool BB = fgt(PhysTime, StartTime, dT_lev / 2);
+  double ndeps = numepss;
+  if (lev < GH->movls)
+    ndeps = numepsb;
+  double TRK4 = PhysTime;
+  int iter_count = 0;
+  int pre = 0, cor = 1;
+  int ERROR = 0;
+  const double dT_mon = dT * pow(0.5, Mymax(0, trfls));
+  const bool need_constraint_after_step =
+      (LastConsOut + dT_mon >= AnasTime) && z4c_constraint_output_due_now();
+
+  if (BH_num > 0 && lev == GH->levels - 1)
+  {
+    if (!z4c_cuda_download_bh_shift_level(GH->PatL[lev], myrank, Sfx0, Sfy0, Sfz0))
+    {
+      if (myrank == 0 && ErrorMonitor->outfile)
+        ErrorMonitor->outfile << "CUDA Z4C failed to download predictor black-hole shift at t = "
+                              << PhysTime << endl;
+      MPI_Abort(MPI_COMM_WORLD, 1);
+    }
+    compute_Porg_rhs(Porg0, Porg_rhs, Sfx0, Sfy0, Sfz0, lev);
+    for (int ithBH = 0; ithBH < BH_num; ithBH++)
+    {
+      f_rungekutta4_scalar(dT_lev, Porg0[ithBH][0], Porg[ithBH][0], Porg_rhs[ithBH][0], iter_count);
+      f_rungekutta4_scalar(dT_lev, Porg0[ithBH][1], Porg[ithBH][1], Porg_rhs[ithBH][1], iter_count);
+      f_rungekutta4_scalar(dT_lev, Porg0[ithBH][2], Porg[ithBH][2], Porg_rhs[ithBH][2], iter_count);
+      if (Symmetry > 0)
+        Porg[ithBH][2] = fabs(Porg[ithBH][2]);
+      if (Symmetry == 2)
+      {
+        Porg[ithBH][0] = fabs(Porg[ithBH][0]);
+        Porg[ithBH][1] = fabs(Porg[ithBH][1]);
+      }
+    }
+  }
+
+  MyList<Patch> *Pp = GH->PatL[lev];
+  while (Pp)
+  {
+    MyList<Block> *BP = Pp->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank)
+      {
+        double *state_in[Z4C_CUDA_STATE_COUNT];
+        double *state_out[Z4C_CUDA_STATE_COUNT];
+        double propspeed[Z4C_CUDA_STATE_COUNT];
+        double soa_flat[3 * Z4C_CUDA_STATE_COUNT];
+        if (!fill_z4c_cuda_views(cg, StateList, state_in, propspeed, soa_flat) ||
+            !fill_z4c_cuda_views(cg, SynchList_pre, state_out))
+        {
+          cout << "CUDA Z4C state list mismatch on predictor step" << endl;
+          MPI_Abort(MPI_COMM_WORLD, 1);
+        }
+        int apply_bam_bc = 0;
+#if (MRBD == 0)
+#if (SommerType == 0)
+        apply_bam_bc = (lev == 0) ? 1 : 0;
+#endif
+#elif (MRBD == 1)
+        apply_bam_bc = 1;
+#endif
+        int keep_resident_state = 1;
+        int apply_enforce_ga = 0;
+#if (AGM == 0)
+        apply_enforce_ga = 1;
+#endif
+        if (z4c_cuda_rk4_substep(cg,
+                                 cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                 state_in, state_out,
+                                 propspeed, soa_flat, Pp->data->bbox,
+                                 dT_lev, TRK4, iter_count, apply_bam_bc,
+                                 Symmetry, lev, ndeps, pre,
+                                 keep_resident_state, apply_enforce_ga, chitiny))
+        {
+          cout << "CUDA Z4C predictor substep failed in domain: ("
+               << cg->bbox[0] << ":" << cg->bbox[3] << ","
+               << cg->bbox[1] << ":" << cg->bbox[4] << ","
+               << cg->bbox[2] << ":" << cg->bbox[5] << ")" << endl;
+          ERROR = 1;
+        }
+      }
+      if (BP == Pp->data->ble)
+        break;
+      BP = BP->next;
+    }
+    Pp = Pp->next;
+  }
+
+  {
+    int erh = ERROR;
+    MPI_Allreduce(&erh, &ERROR, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
+  }
+  if (ERROR)
+  {
+    if (myrank == 0 && ErrorMonitor->outfile)
+      ErrorMonitor->outfile << "CUDA Z4C failed in predictor at t = " << PhysTime
+                            << ", lev = " << lev << endl;
+    MPI_Abort(MPI_COMM_WORLD, 1);
+  }
+
+  {
+    Parallel::AsyncSyncState async_pre;
+    Parallel::Sync_start(GH->PatL[lev], SynchList_pre, Symmetry, sync_cache_pre[lev], async_pre);
+    Parallel::Sync_finish(sync_cache_pre[lev], async_pre, SynchList_pre, Symmetry);
+  }
+
+  if ((lev == a_lev) && (LastAnas + dT_lev >= AnasTime))
+    z4c_cuda_download_level_state(GH->PatL[lev], SynchList_pre, myrank, false);
+  if (lev == a_lev)
+    AnalysisStuff(lev, dT_lev);
+
+  for (iter_count = 1; iter_count < 4; iter_count++)
+  {
+    if (iter_count == 1 || iter_count == 3)
+      TRK4 += dT_lev / 2;
+    Pp = GH->PatL[lev];
+    while (Pp)
+    {
+      MyList<Block> *BP = Pp->data->blb;
+      while (BP)
+      {
+        Block *cg = BP->data;
+        if (myrank == cg->rank)
+        {
+          double *state_in[Z4C_CUDA_STATE_COUNT];
+          double *state_out[Z4C_CUDA_STATE_COUNT];
+          double propspeed[Z4C_CUDA_STATE_COUNT];
+          double soa_flat[3 * Z4C_CUDA_STATE_COUNT];
+          if (!fill_z4c_cuda_views(cg, SynchList_pre, state_in, propspeed, soa_flat) ||
+              !fill_z4c_cuda_views(cg, SynchList_cor, state_out))
+          {
+            cout << "CUDA Z4C state list mismatch on corrector step" << endl;
+            MPI_Abort(MPI_COMM_WORLD, 1);
+          }
+          int apply_bam_bc = 0;
+#if (MRBD == 0)
+#if (SommerType == 0)
+          apply_bam_bc = (lev == 0) ? 1 : 0;
+#endif
+#elif (MRBD == 1)
+          apply_bam_bc = 1;
+#endif
+          int keep_resident_state = 1;
+          int apply_enforce_ga = 0;
+#if (AGM == 0)
+          apply_enforce_ga = 1;
+#elif (AGM == 1)
+          apply_enforce_ga = (iter_count == 3) ? 1 : 0;
+#endif
+          if (z4c_cuda_rk4_substep(cg,
+                                   cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                   state_in, state_out,
+                                   propspeed, soa_flat, Pp->data->bbox,
+                                   dT_lev, TRK4, iter_count, apply_bam_bc,
+                                   Symmetry, lev, ndeps, cor,
+                                   keep_resident_state, apply_enforce_ga, chitiny))
+          {
+            cout << "CUDA Z4C corrector substep failed in domain: ("
+                 << cg->bbox[0] << ":" << cg->bbox[3] << ","
+                 << cg->bbox[1] << ":" << cg->bbox[4] << ","
+                 << cg->bbox[2] << ":" << cg->bbox[5] << ")" << endl;
+            ERROR = 1;
+          }
+          if (!ERROR && iter_count == 3 && need_constraint_after_step)
+          {
+            double *constraints[7] = {
+                cg->fgfs[Cons_Ham->sgfn], cg->fgfs[Cons_Px->sgfn],
+                cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
+                cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn],
+                cg->fgfs[Cons_Gz->sgfn]};
+            double *tz_out[1] = {cg->fgfs[TZ0->sgfn]};
+            const int tz_index = 24;
+            if (z4c_cuda_download_constraint_outputs(cg->shape, constraints) ||
+                z4c_cuda_download_state_subset(cg, cg->shape, 1, &tz_index, tz_out))
+            {
+              cout << "CUDA Z4C constraint download failed in domain: ("
+                   << cg->bbox[0] << ":" << cg->bbox[3] << ","
+                   << cg->bbox[1] << ":" << cg->bbox[4] << ","
+                   << cg->bbox[2] << ":" << cg->bbox[5] << ")" << endl;
+              ERROR = 1;
+            }
+          }
+        }
+        if (BP == Pp->data->ble)
+          break;
+        BP = BP->next;
+      }
+      Pp = Pp->next;
+    }
+
+    {
+      int erh = ERROR;
+      MPI_Allreduce(&erh, &ERROR, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
+    }
+    if (ERROR)
+    {
+      if (myrank == 0 && ErrorMonitor->outfile)
+        ErrorMonitor->outfile << "CUDA Z4C failed in RK4 substep#" << iter_count
+                              << " at t = " << PhysTime
+                              << ", lev = " << lev << endl;
+      MPI_Abort(MPI_COMM_WORLD, 1);
+    }
+
+    {
+      Parallel::AsyncSyncState async_cor;
+      Parallel::Sync_start(GH->PatL[lev], SynchList_cor, Symmetry, sync_cache_cor[lev], async_cor);
+      Parallel::Sync_finish(sync_cache_cor[lev], async_cor, SynchList_cor, Symmetry);
+    }
+
+    if (BH_num > 0 && lev == GH->levels - 1)
+    {
+      if (!z4c_cuda_compute_porg_rhs_resident(GH, lev, myrank, BH_num,
+                                              Porg, Porg1,
+                                              Sfx, Sfy, Sfz, Symmetry))
+      {
+        if (myrank == 0 && ErrorMonitor->outfile)
+          ErrorMonitor->outfile << "CUDA Z4C failed to interpolate black-hole shift at t = "
+                                << PhysTime << endl;
+        MPI_Abort(MPI_COMM_WORLD, 1);
+      }
+      for (int ithBH = 0; ithBH < BH_num; ithBH++)
+      {
+        f_rungekutta4_scalar(dT_lev, Porg0[ithBH][0], Porg1[ithBH][0], Porg_rhs[ithBH][0], iter_count);
+        f_rungekutta4_scalar(dT_lev, Porg0[ithBH][1], Porg1[ithBH][1], Porg_rhs[ithBH][1], iter_count);
+        f_rungekutta4_scalar(dT_lev, Porg0[ithBH][2], Porg1[ithBH][2], Porg_rhs[ithBH][2], iter_count);
+        if (Symmetry > 0)
+          Porg1[ithBH][2] = fabs(Porg1[ithBH][2]);
+        if (Symmetry == 2)
+        {
+          Porg1[ithBH][0] = fabs(Porg1[ithBH][0]);
+          Porg1[ithBH][1] = fabs(Porg1[ithBH][1]);
+        }
+      }
+    }
+
+    if (iter_count < 3)
+    {
+      Pp = GH->PatL[lev];
+      while (Pp)
+      {
+        MyList<Block> *BP = Pp->data->blb;
+        while (BP)
+        {
+          Block *cg = BP->data;
+          cg->swapList(SynchList_pre, SynchList_cor, myrank);
+          if (BP == Pp->data->ble)
+            break;
+          BP = BP->next;
+        }
+        Pp = Pp->next;
+      }
+      if (BH_num > 0 && lev == GH->levels - 1)
+      {
+        for (int ithBH = 0; ithBH < BH_num; ithBH++)
+        {
+          Porg[ithBH][0] = Porg1[ithBH][0];
+          Porg[ithBH][1] = Porg1[ithBH][1];
+          Porg[ithBH][2] = Porg1[ithBH][2];
+        }
+      }
+    }
+  }
+
+  {
+    const bool keep_resident = z4c_cuda_keep_resident_after_step(lev, trfls, a_lev);
+    const bool need_host_after_step =
+        ((lev == a_lev) && (LastAnas + dT_lev >= AnasTime));
+    if (!keep_resident || need_host_after_step)
+      z4c_cuda_download_level_state(GH->PatL[lev], SynchList_cor, myrank, !keep_resident);
+  }
+
+#if (RPS == 0)
+  RestrictProlong(lev, YN, BB);
+#endif
+
+  Pp = GH->PatL[lev];
+  while (Pp)
+  {
+    MyList<Block> *BP = Pp->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      cg->swapList(StateList, SynchList_cor, myrank);
+      cg->swapList(OldStateList, SynchList_cor, myrank);
+      if (BP == Pp->data->ble)
+        break;
+      BP = BP->next;
+    }
+    Pp = Pp->next;
+  }
+  if (BH_num > 0 && lev == GH->levels - 1)
+  {
+    for (int ithBH = 0; ithBH < BH_num; ithBH++)
+    {
+      Porg0[ithBH][0] = Porg1[ithBH][0];
+      Porg0[ithBH][1] = Porg1[ithBH][1];
+      Porg0[ithBH][2] = Porg1[ithBH][2];
+    }
+  }
+#else
   double dT_lev = dT * pow(0.5, Mymax(lev, trfls));
 #ifdef With_AHF
   AH_Step_Find(lev, dT_lev);
@@ -1042,9 +1794,13 @@ void Z4c_class::Step(int lev, int YN)
       Porg0[ithBH][2] = Porg1[ithBH][2];
     }
   }
+#endif
 }
 #else
 // for constraint preserving boundary (CPBC)
+#if USE_CUDA_Z4C && (ABEtype == 2)
+#error "USE_CUDA_Z4C resident path does not support CPBC"
+#endif
 #ifndef WithShell
 #error "CPBC only supports Shell"
 #endif
@@ -2416,8 +3172,14 @@ void Z4c_class::Constraint_Out()
   if (LastConsOut >= AnasTime)
   // Constraint violation
   {
+#if USE_CUDA_Z4C && (ABEtype == 2)
+    bool cuda_constraints_ready = true;
+#else
+    const bool cuda_constraints_ready = false;
+#endif
     // recompute least the constraint data lost for moved new grid
-    for (int lev = 0; lev < GH->levels; lev++)
+    if (!cuda_constraints_ready)
+      for (int lev = 0; lev < GH->levels; lev++)
     {
       // make sure the data consistent for higher levels
       if (lev > 0)

AMSS_NCKU_source/Z4c_rhs.f90

@@ -94,29 +94,31 @@
                Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon,                    &            
                Symmetry,Lev,eps,co)
 
+  if (co == 0) then
 #if (ABV == 0)  
-  call ricci_gamma(ex, X, Y, Z,                                      &
+    call ricci_gamma(ex, X, Y, Z,                                      &
-               chi,                                                  &
+                 chi,                                                  &
-               dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
+                 dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
-               Gamx   ,  Gamy    ,  Gamz    , &
+                 Gamx   ,  Gamy    ,  Gamz    , &
-               Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
+                 Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
-               Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
+                 Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
-               Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
+                 Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
-               Symmetry)
+                 Symmetry)
 #endif
-  call constraint_bssn(ex, X, Y, Z,&
+    call constraint_bssn(ex, X, Y, Z,&
-               chi,trK, &
+                 chi,trK, &
-               dxx,gxy,gxz,dyy,gyz,dzz, &
+                 dxx,gxy,gxz,dyy,gyz,dzz, &
-               Axx,Axy,Axz,Ayy,Ayz,Azz, &
+                 Axx,Axy,Axz,Ayy,Ayz,Azz, &
-               Gamx,Gamy,Gamz,&
+                 Gamx,Gamy,Gamz,&
-               Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
+                 Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
-               Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
+                 Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
-               Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
+                 Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
-               Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
+                 Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
-               Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
+                 Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
-               Symmetry)
+                 Symmetry)
+  endif
 
   return
 
@@ -227,6 +229,7 @@
   call get_Z4cparameters(kappa1,kappa2,kappa3,FF,eta)
 
 !!! sanity check
+#ifdef DEBUG
   dX = sum(chi)+sum(trK)+sum(dxx)+sum(gxy)+sum(gxz)+sum(dyy)+sum(gyz)+sum(dzz) &
       +sum(Axx)+sum(Axy)+sum(Axz)+sum(Ayy)+sum(Ayz)+sum(Azz)                   &
       +sum(Gamx)+sum(Gamy)+sum(Gamz)                                           &
@@ -261,6 +264,7 @@
      gont = 1
      return
   endif
+#endif
 
   PI = dacos(-ONE)
 
@@ -1263,30 +1267,32 @@
 
   endif
 
+  if (co == 0) then
 #if (ABV == 0)  
-  call ricci_gamma(ex, X, Y, Z,                                      &
+    call ricci_gamma(ex, X, Y, Z,                                      &
-               chi,                                                  &
+                 chi,                                                  &
-               dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
+                 dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
-               Gamx   ,  Gamy    ,  Gamz    , &
+                 Gamx   ,  Gamy    ,  Gamz    , &
-               Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
+                 Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
-               Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
+                 Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
-               Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
+                 Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
-               Symmetry)
+                 Symmetry)
 #endif
 
-  call constraint_bssn(ex, X, Y, Z,&
+    call constraint_bssn(ex, X, Y, Z,&
-               chi,trK, &
+                 chi,trK, &
-               dxx,gxy,gxz,dyy,gyz,dzz, &
+                 dxx,gxy,gxz,dyy,gyz,dzz, &
-               Axx,Axy,Axz,Ayy,Ayz,Azz, &
+                 Axx,Axy,Axz,Ayy,Ayz,Azz, &
-               Gamx,Gamy,Gamz,&
+                 Gamx,Gamy,Gamz,&
-               Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
+                 Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
-               Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
+                 Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
-               Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
+                 Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
-               Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
+                 Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
-               Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
+                 Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
-               Symmetry)
+                 Symmetry)
+  endif
 
   gont = 0
 

AMSS_NCKU_source/Z4c_rhs_ss.f90

@@ -122,6 +122,7 @@
   call get_Z4cparameters(kappa1,kappa2,kappa3,FF,eta)
 
 !!! sanity check
+#ifdef DEBUG
   dX = sum(chi)+sum(trK)+sum(dxx)+sum(gxy)+sum(gxz)+sum(dyy)+sum(gyz)+sum(dzz) &
       +sum(Axx)+sum(Axy)+sum(Axz)+sum(Ayy)+sum(Ayz)+sum(Azz)                   &
       +sum(Gamx)+sum(Gamy)+sum(Gamz)                                           &
@@ -156,6 +157,7 @@
      gont = 1
      return
   endif
+#endif
 
   PI = dacos(-ONE)
 
@@ -1388,41 +1390,43 @@
   call kodis_sh(ex,crho,sigma,R,TZ,TZ_rhs,SSS,Symmetry,eps,sst)
   endif
 
+  if (co == 0) then
 #if (ABV == 1)  
-  call ricci_gamma_ss(ex,crho,sigma,R,X, Y, Z,                                 &
+    call ricci_gamma_ss(ex,crho,sigma,R,X, Y, Z,                               &
-               drhodx, drhody, drhodz,                                         &
+                 drhodx, drhody, drhodz,                                       &
-               dsigmadx,dsigmady,dsigmadz,                                     &
+                 dsigmadx,dsigmady,dsigmadz,                                   &
-               dRdx,dRdy,dRdz,                                                 &
+                 dRdx,dRdy,dRdz,                                               &
-               drhodxx,drhodxy,drhodxz,drhodyy,drhodyz,drhodzz,                &
+                 drhodxx,drhodxy,drhodxz,drhodyy,drhodyz,drhodzz,              &
-               dsigmadxx,dsigmadxy,dsigmadxz,dsigmadyy,dsigmadyz,dsigmadzz,    &
+                 dsigmadxx,dsigmadxy,dsigmadxz,dsigmadyy,dsigmadyz,dsigmadzz,  &
-               dRdxx,dRdxy,dRdxz,dRdyy,dRdyz,dRdzz,                            &
+                 dRdxx,dRdxy,dRdxz,dRdyy,dRdyz,dRdzz,                          &
-               chi,                                                  &
+                 chi,                                                          &
-               dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
+                 dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,&
-               Gamx   ,  Gamy    ,  Gamz    , &
+                 Gamx   ,  Gamy    ,  Gamz    , &
-               Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
+                 Gamxxx,Gamxxy,Gamxxz,Gamxyy,Gamxyz,Gamxzz,&
-               Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
+                 Gamyxx,Gamyxy,Gamyxz,Gamyyy,Gamyyz,Gamyzz,&
-               Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
+                 Gamzxx,Gamzxy,Gamzxz,Gamzyy,Gamzyz,Gamzzz,&
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz,&
-               Symmetry,Lev,sst)
+                 Symmetry,Lev,sst)
-  call constraint_bssn_ss(ex,crho,sigma,R,X, Y, Z,  &
-               drhodx, drhody, drhodz,                                         &
-               dsigmadx,dsigmady,dsigmadz,                                     &
-               dRdx,dRdy,dRdz,                                                 &
-               drhodxx,drhodxy,drhodxz,drhodyy,drhodyz,drhodzz,                &
-               dsigmadxx,dsigmadxy,dsigmadxz,dsigmadyy,dsigmadyz,dsigmadzz,    &
-               dRdxx,dRdxy,dRdxz,dRdyy,dRdyz,dRdzz,                            &
-               chi,trK, &
-               dxx,gxy,gxz,dyy,gyz,dzz, &
-               Axx,Axy,Axz,Ayy,Ayz,Azz, &
-               Gamx,Gamy,Gamz,&
-               Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
-               Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
-               Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
-               Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
-               Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
-               Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
-               Symmetry,Lev,sst)
 #endif
+    call constraint_bssn_ss(ex,crho,sigma,R,X, Y, Z,  &
+                 drhodx, drhody, drhodz,                                       &
+                 dsigmadx,dsigmady,dsigmadz,                                   &
+                 dRdx,dRdy,dRdz,                                               &
+                 drhodxx,drhodxy,drhodxz,drhodyy,drhodyz,drhodzz,              &
+                 dsigmadxx,dsigmadxy,dsigmadxz,dsigmadyy,dsigmadyz,dsigmadzz,  &
+                 dRdxx,dRdxy,dRdxz,dRdyy,dRdyz,dRdzz,                          &
+                 chi,trK, &
+                 dxx,gxy,gxz,dyy,gyz,dzz, &
+                 Axx,Axy,Axz,Ayy,Ayz,Azz, &
+                 Gamx,Gamy,Gamz,&
+                 Lap,betax,betay,betaz,rho,Sx,Sy,Sz,&
+                 Gamxxx, Gamxxy, Gamxxz,Gamxyy, Gamxyz, Gamxzz, &
+                 Gamyxx, Gamyxy, Gamyxz,Gamyyy, Gamyyz, Gamyzz, &
+                 Gamzxx, Gamzxy, Gamzxz,Gamzyy, Gamzyz, Gamzzz, &
+                 Rxx,Rxy,Rxz,Ryy,Ryz,Rzz, &
+                 Hcon,Mxcon,Mycon,Mzcon,Gmxcon,Gmycon,Gmzcon, &
+                 Symmetry,Lev,sst)
+  endif
 
   gont = 0
 

AMSS_NCKU_source/bssnEM_class.C

@@ -18,6 +18,9 @@ using namespace std;
 #include "Parallel.h"
 #include "bssnEM_class.h"
 #include "bssn_rhs.h"
+#if USE_CUDA_BSSN
+#include "bssn_rhs_cuda.h"
+#endif
 #include "empart.h"
 #include "initial_puncture.h"
 #include "initial_maxwell.h"
@@ -36,6 +39,106 @@ using namespace std;
 
 //================================================================================================
 
+#if USE_CUDA_BSSN
+namespace {
+
+bool fill_bssn_cuda_views_prefix(Block *cg, MyList<var> *vars,
+                                 double **host_views,
+                                 double *propspeeds = nullptr,
+                                 double *soa_flat = nullptr)
+{
+  int idx = 0;
+  while (vars && idx < BSSN_CUDA_STATE_COUNT)
+  {
+    host_views[idx] = cg->fgfs[vars->data->sgfn];
+    if (propspeeds)
+      propspeeds[idx] = vars->data->propspeed;
+    if (soa_flat)
+    {
+      soa_flat[3 * idx + 0] = vars->data->SoA[0];
+      soa_flat[3 * idx + 1] = vars->data->SoA[1];
+      soa_flat[3 * idx + 2] = vars->data->SoA[2];
+    }
+    vars = vars->next;
+    ++idx;
+  }
+  return idx == BSSN_CUDA_STATE_COUNT;
+}
+
+void skip_bssn_cuda_prefix(MyList<var> *&a, MyList<var> *&b, MyList<var> *&c)
+{
+  for (int i = 0; i < BSSN_CUDA_STATE_COUNT && a && b && c; ++i)
+  {
+    a = a->next;
+    b = b->next;
+    c = c->next;
+  }
+}
+
+void skip_bssn_cuda_prefix(MyList<var> *&a, MyList<var> *&b,
+                           MyList<var> *&c, MyList<var> *&d)
+{
+  for (int i = 0; i < BSSN_CUDA_STATE_COUNT && a && b && c && d; ++i)
+  {
+    a = a->next;
+    b = b->next;
+    c = c->next;
+    d = d->next;
+  }
+}
+
+int run_bssn_em_cuda_substep(Block *cg,
+                             MyList<var> *state_in_list,
+                             MyList<var> *state_out_list,
+                             Patch *patch,
+                             double &dT_lev,
+                             double &TRK4,
+                             int &iter_count,
+                             int &Symmetry,
+                             int lev,
+                             double &ndeps,
+                             int &co,
+                             double &chitiny,
+                             var *rho, var *Sx, var *Sy, var *Sz,
+                             var *Sxx, var *Sxy, var *Sxz,
+                             var *Syy, var *Syz, var *Szz)
+{
+  double *state_in[BSSN_CUDA_STATE_COUNT];
+  double *state_out[BSSN_CUDA_STATE_COUNT];
+  double *matter[BSSN_CUDA_MATTER_COUNT] = {
+      cg->fgfs[rho->sgfn], cg->fgfs[Sx->sgfn], cg->fgfs[Sy->sgfn], cg->fgfs[Sz->sgfn],
+      cg->fgfs[Sxx->sgfn], cg->fgfs[Sxy->sgfn], cg->fgfs[Sxz->sgfn],
+      cg->fgfs[Syy->sgfn], cg->fgfs[Syz->sgfn], cg->fgfs[Szz->sgfn]};
+  double propspeed[BSSN_CUDA_STATE_COUNT];
+  double soa_flat[3 * BSSN_CUDA_STATE_COUNT];
+  if (!fill_bssn_cuda_views_prefix(cg, state_in_list, state_in, propspeed, soa_flat) ||
+      !fill_bssn_cuda_views_prefix(cg, state_out_list, state_out))
+    return 1;
+
+  int apply_bam_bc = 0;
+#if (SommerType == 0)
+#ifndef WithShell
+  apply_bam_bc = (lev == 0) ? 1 : 0;
+#endif
+#endif
+  int use_zero_matter = 0;
+  int keep_resident_state = 0;
+  int apply_enforce_ga = 0;
+  return bssn_cuda_rk4_substep(cg,
+                               cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                               state_in, state_out, matter,
+                               propspeed, soa_flat, patch->bbox,
+                               dT_lev, TRK4, iter_count, apply_bam_bc,
+                               Symmetry, lev, ndeps, co,
+                               use_zero_matter,
+                               keep_resident_state, apply_enforce_ga, chitiny);
+}
+
+}
+#endif
+
+//================================================================================================
+
 // Define bssnEM_class
 
 // It inherits some members and methods from the parent class bssn_class and modifies others.
@@ -244,6 +347,8 @@ void bssnEM_class::Initialize()
     CheckPoint->readcheck_sh(SH, myrank);
 #endif
 
+  Initialize_Level_Runtime();
+
   double h = GH->PatL[0]->data->blb->data->getdX(0);
   for (int i = 1; i < dim; i++)
     h = Mymin(h, GH->PatL[0]->data->blb->data->getdX(i));
@@ -853,6 +958,7 @@ void bssnEM_class::Step(int lev, int YN)
                      cg->fgfs[Ayy0->sgfn], cg->fgfs[Ayz0->sgfn], cg->fgfs[Azz0->sgfn]);
 #endif
 
+        bool used_gpu_substep = false;
         if (
             f_compute_rhs_empart(cg->shape, cg->X[0], cg->X[1], cg->X[2],
                                  cg->fgfs[phi0->sgfn],
@@ -874,7 +980,16 @@ void bssnEM_class::Step(int lev, int YN)
                                  cg->fgfs[Sxx->sgfn], cg->fgfs[Sxy->sgfn], cg->fgfs[Sxz->sgfn],
                                  cg->fgfs[Syy->sgfn], cg->fgfs[Syz->sgfn], cg->fgfs[Szz->sgfn],
                                  Symmetry, lev, ndeps) ||
-            f_compute_rhs_bssn(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
+#if USE_CUDA_BSSN
+            ((used_gpu_substep =
+                  (run_bssn_em_cuda_substep(cg, StateList, SynchList_pre, Pp->data,
+                                            dT_lev, TRK4, iter_count, Symmetry, lev,
+                                            ndeps, pre, chitiny,
+                                            rho, Sx, Sy, Sz, Sxx, Sxy, Sxz, Syy, Syz, Szz) == 0))
+                 ? 0
+                 : 1) ||
+#endif
+            (!used_gpu_substep && f_compute_rhs_bssn(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
                                cg->fgfs[phi0->sgfn], cg->fgfs[trK0->sgfn],
                                cg->fgfs[gxx0->sgfn], cg->fgfs[gxy0->sgfn], cg->fgfs[gxz0->sgfn], 
                                cg->fgfs[gyy0->sgfn], cg->fgfs[gyz0->sgfn], cg->fgfs[gzz0->sgfn],
@@ -907,7 +1022,7 @@ void bssnEM_class::Step(int lev, int YN)
                                cg->fgfs[Cons_Ham->sgfn],
                                cg->fgfs[Cons_Px->sgfn], cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
                                cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn], cg->fgfs[Cons_Gz->sgfn],
-                               Symmetry, lev, ndeps, pre))
+                               Symmetry, lev, ndeps, pre)))
         {
           cout << "find NaN in domain: (" 
                << cg->bbox[0] << ":" << cg->bbox[3] << "," 
@@ -920,6 +1035,10 @@ void bssnEM_class::Step(int lev, int YN)
         {
           MyList<var> *varl0 = StateList, *varl = SynchList_pre, *varlrhs = RHSList;
           // we do not check the correspondence here
+#if USE_CUDA_BSSN
+          if (used_gpu_substep)
+            skip_bssn_cuda_prefix(varl0, varl, varlrhs);
+#endif
 
           while (varl0)
           {
@@ -1221,7 +1340,7 @@ void bssnEM_class::Step(int lev, int YN)
   }
 #endif
 
-  Parallel::Sync(GH->PatL[lev], SynchList_pre, Symmetry);
+  Parallel::Sync_cached(GH->PatL[lev], SynchList_pre, Symmetry, sync_cache_pre[lev]);
 
 #ifdef WithShell
   if (lev == 0)
@@ -1309,6 +1428,7 @@ void bssnEM_class::Step(int lev, int YN)
                          cg->fgfs[Ayy->sgfn], cg->fgfs[Ayz->sgfn], cg->fgfs[Azz->sgfn]);
 #endif
 
+          bool used_gpu_substep = false;
           if (
               f_compute_rhs_empart(cg->shape, cg->X[0], cg->X[1], cg->X[2],
                                    cg->fgfs[phi->sgfn],
@@ -1330,7 +1450,16 @@ void bssnEM_class::Step(int lev, int YN)
                                    cg->fgfs[Sxx->sgfn], cg->fgfs[Sxy->sgfn], cg->fgfs[Sxz->sgfn],
                                    cg->fgfs[Syy->sgfn], cg->fgfs[Syz->sgfn], cg->fgfs[Szz->sgfn],
                                    Symmetry, lev, ndeps) ||
-              f_compute_rhs_bssn(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
+#if USE_CUDA_BSSN
+              ((used_gpu_substep =
+                    (run_bssn_em_cuda_substep(cg, SynchList_pre, SynchList_cor, Pp->data,
+                                              dT_lev, TRK4, iter_count, Symmetry, lev,
+                                              ndeps, cor, chitiny,
+                                              rho, Sx, Sy, Sz, Sxx, Sxy, Sxz, Syy, Syz, Szz) == 0))
+                   ? 0
+                   : 1) ||
+#endif
+              (!used_gpu_substep && f_compute_rhs_bssn(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
                                  cg->fgfs[phi->sgfn], cg->fgfs[trK->sgfn],
                                  cg->fgfs[gxx->sgfn], cg->fgfs[gxy->sgfn], cg->fgfs[gxz->sgfn], 
                                  cg->fgfs[gyy->sgfn], cg->fgfs[gyz->sgfn], cg->fgfs[gzz->sgfn],
@@ -1362,7 +1491,7 @@ void bssnEM_class::Step(int lev, int YN)
                                  cg->fgfs[Cons_Ham->sgfn],
                                  cg->fgfs[Cons_Px->sgfn], cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
                                  cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn], cg->fgfs[Cons_Gz->sgfn],
-                                 Symmetry, lev, ndeps, cor))
+                                 Symmetry, lev, ndeps, cor)))
           {
             cout << "find NaN in domain: (" 
                  << cg->bbox[0] << ":" << cg->bbox[3] << "," 
@@ -1374,6 +1503,10 @@ void bssnEM_class::Step(int lev, int YN)
           {
             MyList<var> *varl0 = StateList, *varl = SynchList_pre, *varl1 = SynchList_cor, *varlrhs = RHSList;
             // we do not check the correspondence here
+#if USE_CUDA_BSSN
+            if (used_gpu_substep)
+              skip_bssn_cuda_prefix(varl0, varl, varl1, varlrhs);
+#endif
 
             while (varl0)
             {
@@ -1683,7 +1816,7 @@ void bssnEM_class::Step(int lev, int YN)
     }
 #endif
 
-    Parallel::Sync(GH->PatL[lev], SynchList_cor, Symmetry);
+    Parallel::Sync_cached(GH->PatL[lev], SynchList_cor, Symmetry, sync_cache_cor[lev]);
 
 #ifdef WithShell
     if (lev == 0)

AMSS_NCKU_source/bssnEScalar_class.C

@@ -18,6 +18,9 @@ using namespace std;
 #include "Parallel.h"
 #include "bssnEScalar_class.h"
 #include "bssn_rhs.h"
+#if USE_CUDA_BSSN
+#include "bssn_rhs_cuda.h"
+#endif
 #include "initial_puncture.h"
 #include "enforce_algebra.h"
 #include "rungekutta4_rout.h"
@@ -33,6 +36,350 @@ using namespace std;
 
 //================================================================================================
 
+namespace {
+
+int amss_escalar_analysis_map_every()
+{
+  static int every = -1;
+  if (every < 0)
+  {
+    const char *env = getenv("AMSS_ANALYSIS_MAP_EVERY");
+    every = (env && atoi(env) > 0) ? atoi(env) : 1;
+  }
+  return every;
+}
+
+}
+
+//================================================================================================
+
+#if USE_CUDA_BSSN
+extern "C" {
+#ifdef fortran1
+void set_escalar_parameter(double &, double &, double &, double &, double &);
+#endif
+#ifdef fortran2
+void SET_ESCALAR_PARAMETER(double &, double &, double &, double &, double &);
+#endif
+#ifdef fortran3
+void set_escalar_parameter_(double &, double &, double &, double &, double &);
+#endif
+}
+
+namespace {
+
+bool fill_bssn_cuda_views_prefix(Block *cg, MyList<var> *vars,
+                                 double **host_views,
+                                 double *propspeeds = nullptr,
+                                 double *soa_flat = nullptr)
+{
+  int idx = 0;
+  while (vars && idx < BSSN_CUDA_STATE_COUNT)
+  {
+    host_views[idx] = cg->fgfs[vars->data->sgfn];
+    if (propspeeds)
+      propspeeds[idx] = vars->data->propspeed;
+    if (soa_flat)
+    {
+      soa_flat[3 * idx + 0] = vars->data->SoA[0];
+      soa_flat[3 * idx + 1] = vars->data->SoA[1];
+      soa_flat[3 * idx + 2] = vars->data->SoA[2];
+    }
+    vars = vars->next;
+    ++idx;
+  }
+  return idx == BSSN_CUDA_STATE_COUNT;
+}
+
+void skip_bssn_cuda_prefix(MyList<var> *&a, MyList<var> *&b, MyList<var> *&c)
+{
+  for (int i = 0; i < BSSN_CUDA_STATE_COUNT && a && b && c; ++i)
+  {
+    a = a->next;
+    b = b->next;
+    c = c->next;
+  }
+}
+
+void skip_bssn_cuda_prefix(MyList<var> *&a, MyList<var> *&b,
+                           MyList<var> *&c, MyList<var> *&d)
+{
+  for (int i = 0; i < BSSN_CUDA_STATE_COUNT && a && b && c && d; ++i)
+  {
+    a = a->next;
+    b = b->next;
+    c = c->next;
+    d = d->next;
+  }
+}
+
+MyList<var> *clone_var_list_prefix(MyList<var> *src, int count)
+{
+  MyList<var> *dst = nullptr;
+  MyList<var> *tail = nullptr;
+  for (int i = 0; i < count && src; ++i, src = src->next)
+  {
+    MyList<var> *node = new MyList<var>(src->data);
+    if (!dst)
+      dst = node;
+    else
+      tail->next = node;
+    tail = node;
+  }
+  return dst;
+}
+
+bool escalar_gpu_rk_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_ESCALAR_GPU_RK");
+    enabled = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+bool escalar_resident_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_ESCALAR_RESIDENT");
+    const char *experimental = getenv("AMSS_ESCALAR_RESIDENT_EXPERIMENTAL");
+    enabled = (env && atoi(env) != 0 &&
+               experimental && atoi(experimental) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+bool escalar_step_profile_enabled()
+{
+  static int enabled = -1;
+  if (enabled < 0)
+  {
+    const char *env = getenv("AMSS_ESCALAR_STEP_PROFILE");
+    enabled = (env && atoi(env) != 0) ? 1 : 0;
+  }
+  return enabled != 0;
+}
+
+int escalar_step_profile_every()
+{
+  static int every = -1;
+  if (every < 0)
+  {
+    const char *env = getenv("AMSS_ESCALAR_STEP_PROFILE_EVERY");
+    every = (env && atoi(env) > 0) ? atoi(env) : 1;
+  }
+  return every;
+}
+
+struct EScalarStepProfile
+{
+  double start;
+  double predictor_rhs;
+  double predictor_sync;
+  double analysis;
+  double corrector_rhs;
+  double corrector_sync;
+  double restrict_prolong;
+  double other_sync;
+};
+
+void escalar_profile_init(EScalarStepProfile &p)
+{
+  p.start = MPI_Wtime();
+  p.predictor_rhs = 0.0;
+  p.predictor_sync = 0.0;
+  p.analysis = 0.0;
+  p.corrector_rhs = 0.0;
+  p.corrector_sync = 0.0;
+  p.restrict_prolong = 0.0;
+  p.other_sync = 0.0;
+}
+
+void escalar_profile_add(double &bucket, double t0)
+{
+  bucket += MPI_Wtime() - t0;
+}
+
+void escalar_profile_report(const EScalarStepProfile &p, int lev, int myrank)
+{
+  if (myrank != 0 || !escalar_step_profile_enabled())
+    return;
+  static long long call_count = 0;
+  ++call_count;
+  const int every = escalar_step_profile_every();
+  if (every > 1 && (call_count % every) != 0)
+    return;
+  const double total = MPI_Wtime() - p.start;
+  fprintf(stderr,
+          "[AMSS-ESCALAR-PROFILE] call=%lld lev=%d total=%.6f pred_rhs=%.6f pred_sync=%.6f analysis=%.6f corr_rhs=%.6f corr_sync=%.6f rp=%.6f other_sync=%.6f\n",
+          call_count, lev, total, p.predictor_rhs, p.predictor_sync,
+          p.analysis, p.corrector_rhs, p.corrector_sync,
+          p.restrict_prolong, p.other_sync);
+  fflush(stderr);
+}
+
+void clear_var_list(MyList<var> *&list)
+{
+  if (list)
+  {
+    list->clearList();
+    list = nullptr;
+  }
+}
+
+void download_bssn_cuda_prefix_if_present(MyList<Patch> *PatL,
+                                          MyList<var> *vars,
+                                          int myrank)
+{
+  while (PatL)
+  {
+    MyList<Block> *BP = PatL->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank)
+      {
+        double *views[BSSN_CUDA_STATE_COUNT];
+        if (fill_bssn_cuda_views_prefix(cg, vars, views))
+          bssn_cuda_download_resident_state_if_present(cg, cg->shape, views);
+      }
+      if (BP == PatL->data->ble)
+        break;
+      BP = BP->next;
+    }
+    PatL = PatL->next;
+  }
+}
+
+void download_escalar_cuda_pair_if_present(MyList<Patch> *PatL,
+                                           var *Sphi_var,
+                                           var *Spi_var,
+                                           int myrank)
+{
+  if (!Sphi_var || !Spi_var)
+    return;
+  while (PatL)
+  {
+    MyList<Block> *BP = PatL->data->blb;
+    while (BP)
+    {
+      Block *cg = BP->data;
+      if (myrank == cg->rank)
+      {
+        bssn_cuda_escalar_download_fields_if_present(
+            cg, cg->shape,
+            cg->fgfs[Sphi_var->sgfn],
+            cg->fgfs[Spi_var->sgfn]);
+      }
+      if (BP == PatL->data->ble)
+        break;
+      BP = BP->next;
+    }
+    PatL = PatL->next;
+  }
+}
+
+int run_bssn_escalar_cuda_substep(Block *cg,
+                                  MyList<var> *state_in_list,
+                                  MyList<var> *state_out_list,
+                                  Patch *patch,
+                                  double &dT_lev,
+                                  double &TRK4,
+                                  int &iter_count,
+                                  int &Symmetry,
+                                  int lev,
+                                  double &ndeps,
+                                  int &co,
+                                  double &chitiny,
+                                  var *Sphi_in, var *Spi_in,
+                                  var *Sphi_out, var *Spi_out,
+                                  var *Sphi_rhs, var *Spi_rhs,
+                                  var *rho, var *Sx, var *Sy, var *Sz,
+                                  var *Sxx, var *Sxy, var *Sxz,
+                                  var *Syy, var *Syz, var *Szz)
+{
+  double *state_in[BSSN_CUDA_STATE_COUNT];
+  double *state_out[BSSN_CUDA_STATE_COUNT];
+  double propspeed[BSSN_CUDA_STATE_COUNT];
+  double soa_flat[3 * BSSN_CUDA_STATE_COUNT];
+  if (!fill_bssn_cuda_views_prefix(cg, state_in_list, state_in, propspeed, soa_flat) ||
+      !fill_bssn_cuda_views_prefix(cg, state_out_list, state_out))
+    return 1;
+
+  double a2 = 0.0, phi0 = 0.0, r0 = 0.0, sigma0 = 0.0, l2 = 0.0;
+#ifdef fortran1
+  set_escalar_parameter(a2, phi0, r0, sigma0, l2);
+#endif
+#ifdef fortran2
+  SET_ESCALAR_PARAMETER(a2, phi0, r0, sigma0, l2);
+#endif
+#ifdef fortran3
+  set_escalar_parameter_(a2, phi0, r0, sigma0, l2);
+#endif
+  int apply_enforce_ga = 0;
+#if (AGM == 0)
+  apply_enforce_ga = 1;
+#elif (AGM == 1)
+  apply_enforce_ga = (iter_count == 3) ? 1 : 0;
+#endif
+  if (bssn_cuda_compute_escalar_matter(cg,
+                                       cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                       state_in,
+                                       cg->fgfs[Sphi_in->sgfn],
+                                       cg->fgfs[Spi_in->sgfn],
+                                       cg->fgfs[Sphi_rhs->sgfn],
+                                       cg->fgfs[Spi_rhs->sgfn],
+                                       a2, Symmetry, lev, ndeps, co, apply_enforce_ga))
+    return 1;
+
+  int apply_bam_bc = 0;
+#if (SommerType == 0)
+#ifndef WithShell
+  apply_bam_bc = (lev == 0) ? 1 : 0;
+#endif
+#endif
+  if (escalar_gpu_rk_enabled())
+  {
+    double scalar_propspeed[2] = {
+      Sphi_in->propspeed, Spi_in->propspeed
+    };
+    double scalar_soa[6] = {
+      Sphi_in->SoA[0], Sphi_in->SoA[1], Sphi_in->SoA[2],
+      Spi_in->SoA[0], Spi_in->SoA[1], Spi_in->SoA[2]
+    };
+    if (bssn_cuda_escalar_finalize_scalar_fields(cg,
+                                                 cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                                 cg->fgfs[Sphi_out->sgfn],
+                                                 cg->fgfs[Spi_out->sgfn],
+                                                 scalar_propspeed,
+                                                 scalar_soa,
+                                                 patch->bbox,
+                                                 dT_lev, iter_count, apply_bam_bc,
+                                                 Symmetry, lev, ndeps, co))
+      return 1;
+  }
+  int use_zero_matter = 0;
+  int keep_resident_state = 1;
+  double **matter_precomputed = nullptr;
+  return bssn_cuda_rk4_substep(cg,
+                               cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                               state_in, state_out, matter_precomputed,
+                               propspeed, soa_flat, patch->bbox,
+                               dT_lev, TRK4, iter_count, apply_bam_bc,
+                               Symmetry, lev, ndeps, co,
+                               use_zero_matter,
+                               keep_resident_state, apply_enforce_ga, chitiny);
+}
+
+}
+#endif
+
+//================================================================================================
+
 // Define bssnEScalar_class
 
 // It inherits some members and methods from the parent class bssn_class and modifies others.
@@ -52,6 +399,14 @@ bssnEScalar_class::bssnEScalar_class(double Couranti, double StartTimei, double
                                                   Symmetryi, checkruni, checkfilenamei, numepssi, numepsbi, numepshi,
                                                   a_levi, maxli, decni, maxrexi, drexi)
 {
+  BSSNStateList = nullptr;
+  BSSNSynchList_pre = nullptr;
+  BSSNSynchList_cor = nullptr;
+  ScalarSynchList_pre = nullptr;
+  ScalarSynchList_cor = nullptr;
+  sync_cache_scalar_pre = nullptr;
+  sync_cache_scalar_cor = nullptr;
+
   // setup Monitors
   {
     char str[50];
@@ -110,6 +465,16 @@ void bssnEScalar_class::Initialize()
   DumpList->insert(Spi0);
   DumpList->insert(Cons_fR);
 
+#if USE_CUDA_BSSN
+  BSSNStateList = clone_var_list_prefix(StateList, BSSN_CUDA_STATE_COUNT);
+  BSSNSynchList_pre = clone_var_list_prefix(SynchList_pre, BSSN_CUDA_STATE_COUNT);
+  BSSNSynchList_cor = clone_var_list_prefix(SynchList_cor, BSSN_CUDA_STATE_COUNT);
+  ScalarSynchList_pre = new MyList<var>(Sphi);
+  ScalarSynchList_pre->insert(Spi);
+  ScalarSynchList_cor = new MyList<var>(Sphi1);
+  ScalarSynchList_cor->insert(Spi1);
+#endif
+
   CheckPoint->addvariablelist(StateList);
   CheckPoint->addvariablelist(OldStateList);
   
@@ -151,6 +516,14 @@ void bssnEScalar_class::Initialize()
     CheckPoint->readcheck_sh(SH, myrank);
 #endif
 
+  Initialize_Level_Runtime();
+#if USE_CUDA_BSSN
+  if (!sync_cache_scalar_pre)
+    sync_cache_scalar_pre = new Parallel::SyncCache[GH->levels];
+  if (!sync_cache_scalar_cor)
+    sync_cache_scalar_cor = new Parallel::SyncCache[GH->levels];
+#endif
+
   double h = GH->PatL[0]->data->blb->data->getdX(0);
   for (int i = 1; i < dim; i++)
     h = Mymin(h, GH->PatL[0]->data->blb->data->getdX(i));
@@ -179,6 +552,30 @@ void bssnEScalar_class::Initialize()
 
 bssnEScalar_class::~bssnEScalar_class()
 {
+#if USE_CUDA_BSSN
+  clear_var_list(BSSNStateList);
+  clear_var_list(BSSNSynchList_pre);
+  clear_var_list(BSSNSynchList_cor);
+  clear_var_list(ScalarSynchList_pre);
+  clear_var_list(ScalarSynchList_cor);
+  if (sync_cache_scalar_pre)
+  {
+    const int levels = GH ? GH->levels : 0;
+    for (int i = 0; i < levels; ++i)
+      sync_cache_scalar_pre[i].destroy();
+    delete[] sync_cache_scalar_pre;
+    sync_cache_scalar_pre = nullptr;
+  }
+  if (sync_cache_scalar_cor)
+  {
+    const int levels = GH ? GH->levels : 0;
+    for (int i = 0; i < levels; ++i)
+      sync_cache_scalar_cor[i].destroy();
+    delete[] sync_cache_scalar_cor;
+    sync_cache_scalar_cor = nullptr;
+  }
+#endif
+
   delete Sphio;
   delete Spio;
   delete Sphi0;
@@ -719,27 +1116,44 @@ void bssnEScalar_class::Step(int lev, int YN)
   int iter_count = 0; // count RK4 substeps
   int pre = 0, cor = 1;
   int ERROR = 0;
+  EScalarStepProfile escalar_profile;
+  escalar_profile_init(escalar_profile);
 
   MyList<ss_patch> *sPp;
   // Predictor
+  const double escalar_profile_predictor_rhs_start = MPI_Wtime();
   MyList<Patch> *Pp = GH->PatL[lev];
   while (Pp)
   {
     MyList<Block> *BP = Pp->data->blb;
     while (BP)
-    {
-      Block *cg = BP->data;
-      if (myrank == cg->rank)
       {
+        Block *cg = BP->data;
+        if (myrank == cg->rank)
+        {
+#if !USE_CUDA_BSSN
 #if (AGM == 0)
         f_enforce_ga(cg->shape,
                      cg->fgfs[gxx0->sgfn], cg->fgfs[gxy0->sgfn], cg->fgfs[gxz0->sgfn],
                      cg->fgfs[gyy0->sgfn], cg->fgfs[gyz0->sgfn], cg->fgfs[gzz0->sgfn],
                      cg->fgfs[Axx0->sgfn], cg->fgfs[Axy0->sgfn], cg->fgfs[Axz0->sgfn],
                      cg->fgfs[Ayy0->sgfn], cg->fgfs[Ayz0->sgfn], cg->fgfs[Azz0->sgfn]);
+#endif
 #endif
 
-        if (f_compute_rhs_bssn_escalar(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
+        bool used_gpu_substep = false;
+        if (
+#if USE_CUDA_BSSN
+            ((used_gpu_substep =
+                  (run_bssn_escalar_cuda_substep(cg, StateList, SynchList_pre, Pp->data,
+                                                 dT_lev, TRK4, iter_count, Symmetry, lev,
+                                                 ndeps, pre, chitiny,
+                                                 Sphi0, Spi0, Sphi, Spi, Sphi_rhs, Spi_rhs,
+                                                 rho, Sx, Sy, Sz, Sxx, Sxy, Sxz, Syy, Syz, Szz) == 0))
+                 ? 0
+                 : 1) ||
+#endif
+            (!used_gpu_substep && f_compute_rhs_bssn_escalar(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
                                        cg->fgfs[phi0->sgfn], cg->fgfs[trK0->sgfn],
                                        cg->fgfs[gxx0->sgfn], cg->fgfs[gxy0->sgfn], cg->fgfs[gxz0->sgfn], 
                                        cg->fgfs[gyy0->sgfn], cg->fgfs[gyz0->sgfn], cg->fgfs[gzz0->sgfn],
@@ -774,7 +1188,7 @@ void bssnEScalar_class::Step(int lev, int YN)
                                        cg->fgfs[Cons_Ham->sgfn],
                                        cg->fgfs[Cons_Px->sgfn], cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
                                        cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn], cg->fgfs[Cons_Gz->sgfn],
-                                       Symmetry, lev, ndeps, pre))
+                                       Symmetry, lev, ndeps, pre)))
         {
           cout << "find NaN in domain: (" 
                << cg->bbox[0] << ":" << cg->bbox[3] << "," 
@@ -786,8 +1200,40 @@ void bssnEScalar_class::Step(int lev, int YN)
         // rk4 substep and boundary
         {
           MyList<var> *varl0 = StateList, *varl = SynchList_pre, *varlrhs = RHSList; // we do not check the correspondence here
+#if USE_CUDA_BSSN
+          if (used_gpu_substep)
+            skip_bssn_cuda_prefix(varl0, varl, varlrhs);
+#endif
+          const bool scalar_gpu_rk_done =
+#if USE_CUDA_BSSN
+            used_gpu_substep && escalar_gpu_rk_enabled();
+#else
+            false;
+#endif
           while (varl0)
           {
+            if (scalar_gpu_rk_done)
+            {
+              if (!escalar_resident_enabled())
+              {
+#ifndef WithShell
+              if (lev > 0) // fix BD point
+#endif
+                f_sommerfeld_rout(cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                  Pp->data->bbox[0], Pp->data->bbox[1], Pp->data->bbox[2],
+                                  Pp->data->bbox[3], Pp->data->bbox[4], Pp->data->bbox[5],
+                                  dT_lev, cg->fgfs[phi0->sgfn],
+                                  cg->fgfs[Lap0->sgfn],
+                                  cg->fgfs[varl0->data->sgfn], cg->fgfs[varl->data->sgfn],
+                                  varl0->data->SoA,
+                                  Symmetry, cor);
+              }
+
+              varl0 = varl0->next;
+              varl = varl->next;
+              varlrhs = varlrhs->next;
+              continue;
+            }
 #ifndef WithShell
             if (lev == 0) // sommerfeld indeed
               f_sommerfeld_routbam(cg->shape, cg->X[0], cg->X[1], cg->X[2],
@@ -821,7 +1267,8 @@ void bssnEScalar_class::Step(int lev, int YN)
             varlrhs = varlrhs->next;
           }
         }
-        f_lowerboundset(cg->shape, cg->fgfs[phi->sgfn], chitiny);
+        if (!used_gpu_substep)
+          f_lowerboundset(cg->shape, cg->fgfs[phi->sgfn], chitiny);
       }
       if (BP == Pp->data->ble)
         break;
@@ -829,6 +1276,7 @@ void bssnEScalar_class::Step(int lev, int YN)
     }
     Pp = Pp->next;
   }
+  escalar_profile_add(escalar_profile.predictor_rhs, escalar_profile_predictor_rhs_start);
   // check error information
   {
     int erh = ERROR;
@@ -993,7 +1441,16 @@ void bssnEScalar_class::Step(int lev, int YN)
   }
 #endif
 
-  Parallel::Sync(GH->PatL[lev], SynchList_pre, Symmetry);
+#if USE_CUDA_BSSN
+  const double escalar_profile_predictor_sync_start = MPI_Wtime();
+  Parallel::Sync_cached(GH->PatL[lev], BSSNSynchList_pre, Symmetry, sync_cache_pre[lev]);
+  Parallel::Sync_cached(GH->PatL[lev], ScalarSynchList_pre, Symmetry, sync_cache_scalar_pre[lev]);
+  escalar_profile_add(escalar_profile.predictor_sync, escalar_profile_predictor_sync_start);
+#else
+  const double escalar_profile_predictor_sync_start = MPI_Wtime();
+  Parallel::Sync_cached(GH->PatL[lev], SynchList_pre, Symmetry, sync_cache_pre[lev]);
+  escalar_profile_add(escalar_profile.predictor_sync, escalar_profile_predictor_sync_start);
+#endif
 
 #ifdef WithShell
   if (lev == 0)
@@ -1049,7 +1506,13 @@ void bssnEScalar_class::Step(int lev, int YN)
   // Warning NOTE: the variables1 are used as temp storege room
   if (lev == a_lev)
   {
+    const double escalar_profile_analysis_start = MPI_Wtime();
+#if USE_CUDA_BSSN
+    if (escalar_resident_enabled())
+      download_escalar_cuda_pair_if_present(GH->PatL[lev], Sphi, Spi, myrank);
+#endif
     AnalysisStuff_EScalar(lev, dT_lev);
+    escalar_profile_add(escalar_profile.analysis, escalar_profile_analysis_start);
   }
   // corrector
   for (iter_count = 1; iter_count < 4; iter_count++)
@@ -1057,6 +1520,7 @@ void bssnEScalar_class::Step(int lev, int YN)
     // for RK4: t0, t0+dt/2, t0+dt/2, t0+dt;
     if (iter_count == 1 || iter_count == 3)
       TRK4 += dT_lev / 2;
+    const double escalar_profile_corrector_rhs_start = MPI_Wtime();
     Pp = GH->PatL[lev];
     while (Pp)
     {
@@ -1066,6 +1530,7 @@ void bssnEScalar_class::Step(int lev, int YN)
         Block *cg = BP->data;
         if (myrank == cg->rank)
         {
+#if !USE_CUDA_BSSN
 #if (AGM == 0)
           f_enforce_ga(cg->shape,
                        cg->fgfs[gxx->sgfn], cg->fgfs[gxy->sgfn], cg->fgfs[gxz->sgfn],
@@ -1079,9 +1544,22 @@ void bssnEScalar_class::Step(int lev, int YN)
                          cg->fgfs[gyy->sgfn], cg->fgfs[gyz->sgfn], cg->fgfs[gzz->sgfn],
                          cg->fgfs[Axx->sgfn], cg->fgfs[Axy->sgfn], cg->fgfs[Axz->sgfn],
                          cg->fgfs[Ayy->sgfn], cg->fgfs[Ayz->sgfn], cg->fgfs[Azz->sgfn]);
+#endif
 #endif
 
-          if (f_compute_rhs_bssn_escalar(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
+          bool used_gpu_substep = false;
+          if (
+#if USE_CUDA_BSSN
+              ((used_gpu_substep =
+                    (run_bssn_escalar_cuda_substep(cg, SynchList_pre, SynchList_cor, Pp->data,
+                                                   dT_lev, TRK4, iter_count, Symmetry, lev,
+                                                   ndeps, cor, chitiny,
+                                                   Sphi, Spi, Sphi1, Spi1, Sphi_rhs, Spi_rhs,
+                                                   rho, Sx, Sy, Sz, Sxx, Sxy, Sxz, Syy, Syz, Szz) == 0))
+                   ? 0
+                   : 1) ||
+#endif
+              (!used_gpu_substep && f_compute_rhs_bssn_escalar(cg->shape, TRK4, cg->X[0], cg->X[1], cg->X[2],
                                          cg->fgfs[phi->sgfn], cg->fgfs[trK->sgfn],
                                          cg->fgfs[gxx->sgfn], cg->fgfs[gxy->sgfn], cg->fgfs[gxz->sgfn], 
                                          cg->fgfs[gyy->sgfn], cg->fgfs[gyz->sgfn], cg->fgfs[gzz->sgfn],
@@ -1117,7 +1595,7 @@ void bssnEScalar_class::Step(int lev, int YN)
                                          cg->fgfs[Cons_Ham->sgfn],
                                          cg->fgfs[Cons_Px->sgfn], cg->fgfs[Cons_Py->sgfn], cg->fgfs[Cons_Pz->sgfn],
                                          cg->fgfs[Cons_Gx->sgfn], cg->fgfs[Cons_Gy->sgfn], cg->fgfs[Cons_Gz->sgfn],
-                                         Symmetry, lev, ndeps, cor))
+                                         Symmetry, lev, ndeps, cor)))
           {
             cout << "find NaN in domain: (" 
                  << cg->bbox[0] << ":" << cg->bbox[3] << "," 
@@ -1129,9 +1607,42 @@ void bssnEScalar_class::Step(int lev, int YN)
           {
             MyList<var> *varl0 = StateList, *varl = SynchList_pre, *varl1 = SynchList_cor, *varlrhs = RHSList;
             // we do not check the correspondence here
+#if USE_CUDA_BSSN
+            if (used_gpu_substep)
+              skip_bssn_cuda_prefix(varl0, varl, varl1, varlrhs);
+#endif
+            const bool scalar_gpu_rk_done =
+#if USE_CUDA_BSSN
+              used_gpu_substep && escalar_gpu_rk_enabled();
+#else
+              false;
+#endif
 
             while (varl0)
             {
+              if (scalar_gpu_rk_done)
+              {
+                if (!escalar_resident_enabled())
+                {
+#ifndef WithShell
+                if (lev > 0) // fix BD point
+#endif
+                  f_sommerfeld_rout(cg->shape, cg->X[0], cg->X[1], cg->X[2],
+                                    Pp->data->bbox[0], Pp->data->bbox[1], Pp->data->bbox[2],
+                                    Pp->data->bbox[3], Pp->data->bbox[4], Pp->data->bbox[5],
+                                    dT_lev, cg->fgfs[phi0->sgfn],
+                                    cg->fgfs[Lap0->sgfn],
+                                    cg->fgfs[varl0->data->sgfn], cg->fgfs[varl1->data->sgfn],
+                                    varl0->data->SoA,
+                                    Symmetry, cor);
+                }
+
+                varl0 = varl0->next;
+                varl = varl->next;
+                varl1 = varl1->next;
+                varlrhs = varlrhs->next;
+                continue;
+              }
 #ifndef WithShell
               if (lev == 0) // sommerfeld indeed
                 f_sommerfeld_routbam(cg->shape, cg->X[0], cg->X[1], cg->X[2],
@@ -1166,7 +1677,8 @@ void bssnEScalar_class::Step(int lev, int YN)
               varlrhs = varlrhs->next;
             }
           }
-          f_lowerboundset(cg->shape, cg->fgfs[phi1->sgfn], chitiny);
+          if (!used_gpu_substep)
+            f_lowerboundset(cg->shape, cg->fgfs[phi1->sgfn], chitiny);
         }
         if (BP == Pp->data->ble)
           break;
@@ -1174,6 +1686,7 @@ void bssnEScalar_class::Step(int lev, int YN)
       }
       Pp = Pp->next;
     }
+    escalar_profile_add(escalar_profile.corrector_rhs, escalar_profile_corrector_rhs_start);
 
     // check error information
     {
@@ -1349,7 +1862,16 @@ void bssnEScalar_class::Step(int lev, int YN)
     }
 #endif
 
-    Parallel::Sync(GH->PatL[lev], SynchList_cor, Symmetry);
+#if USE_CUDA_BSSN
+    const double escalar_profile_corrector_sync_start = MPI_Wtime();
+    Parallel::Sync_cached(GH->PatL[lev], BSSNSynchList_cor, Symmetry, sync_cache_cor[lev]);
+    Parallel::Sync_cached(GH->PatL[lev], ScalarSynchList_cor, Symmetry, sync_cache_scalar_cor[lev]);
+    escalar_profile_add(escalar_profile.corrector_sync, escalar_profile_corrector_sync_start);
+#else
+    const double escalar_profile_corrector_sync_start = MPI_Wtime();
+    Parallel::Sync_cached(GH->PatL[lev], SynchList_cor, Symmetry, sync_cache_cor[lev]);
+    escalar_profile_add(escalar_profile.corrector_sync, escalar_profile_corrector_sync_start);
+#endif
 
 #ifdef WithShell
     if (lev == 0)
@@ -1451,7 +1973,21 @@ void bssnEScalar_class::Step(int lev, int YN)
 
 #if (RPS == 0)
   // mesh refinement boundary part
+  const double escalar_profile_rp_start = MPI_Wtime();
+#if USE_CUDA_BSSN
+  {
+    const char *mixed_env = getenv("AMSS_ESCALAR_MIXED_GPU_RP");
+    const bool mixed_gpu_rp = (mixed_env && atoi(mixed_env) != 0);
+    const char *split_env = getenv("AMSS_ESCALAR_SPLIT_RP");
+    const bool split_rp = (split_env && atoi(split_env) != 0);
+    if (escalar_resident_enabled() && !split_rp)
+      download_escalar_cuda_pair_if_present(GH->PatL[lev], Sphi1, Spi1, myrank);
+    if (!mixed_gpu_rp && !split_rp)
+      download_bssn_cuda_prefix_if_present(GH->PatL[lev], SynchList_cor, myrank);
+  }
+#endif
   RestrictProlong(lev, YN, BB);
+  escalar_profile_add(escalar_profile.restrict_prolong, escalar_profile_rp_start);
 
 #ifdef WithShell
   if (lev == 0)
@@ -1523,6 +2059,7 @@ void bssnEScalar_class::Step(int lev, int YN)
       Porg0[ithBH][2] = Porg1[ithBH][2];
     }
   }
+  escalar_profile_report(escalar_profile, lev, myrank);
 }
 
 //================================================================================================
@@ -2060,6 +2597,23 @@ void bssnEScalar_class::Constraint_Out()
   if (LastConsOut >= AnasTime)
   // Constraint violation
   {
+    const int constraint_map_every = amss_escalar_analysis_map_every();
+    static long long constraint_map_counter = 0;
+    const bool refresh_constraints =
+        constraint_map_every <= 1 ||
+        (constraint_map_counter % constraint_map_every) == 0;
+    constraint_map_counter++;
+    if (!refresh_constraints)
+    {
+      LastConsOut = 0;
+      return;
+    }
+
+#if USE_CUDA_BSSN
+    for (int lev = 0; lev < GH->levels; lev++)
+      download_bssn_cuda_prefix_if_present(GH->PatL[lev], StateList, myrank);
+#endif
+
     // recompute least the constraint data lost for moved new grid
     for (int lev = 0; lev < GH->levels; lev++)
     {

AMSS_NCKU_source/bssnEScalar_class.h

@@ -63,6 +63,10 @@ protected:
 
      var *Cons_fR;
 
+     MyList<var> *BSSNStateList, *BSSNSynchList_pre, *BSSNSynchList_cor;
+     MyList<var> *ScalarSynchList_pre, *ScalarSynchList_cor;
+     Parallel::SyncCache *sync_cache_scalar_pre, *sync_cache_scalar_cor;
+
      monitor *MaxScalar_Monitor;
 };
 

AMSS_NCKU_source/bssnEScalar_rhs.f90

@@ -5,6 +5,138 @@
 
 #include "macrodef.fh"
 
+! scalar RHS and stress-energy only; BSSN RHS can be supplied by CUDA.
+  function compute_rhs_bssn_escalar_matter(ex, T, X, Y, Z,                    &
+               chi    ,   trK    ,                                            &
+               dxx    ,   gxy    ,   gxz    ,   dyy    ,   gyz    ,   dzz,    &
+               Axx    ,   Axy    ,   Axz    ,   Ayy    ,   Ayz    ,   Azz,    &
+               Gamx   ,  Gamy    ,  Gamz    ,                                 &
+               Lap    ,  betax   ,  betay   ,  betaz   ,                      &
+               dtSfx  ,  dtSfy   ,  dtSfz   ,                                 &
+               Sphi   ,   Spi    ,                                            &
+               Sphi_rhs ,   Spi_rhs ,                                         &
+               rho,Sx,Sy,Sz,Sxx,Sxy,Sxz,Syy,Syz,Szz,                          &
+               Symmetry,Lev,eps) result(gont)
+  implicit none
+
+  integer,intent(in ):: ex(1:3), Symmetry,Lev
+  real*8, intent(in ):: T
+  real*8, intent(in ):: X(1:ex(1)),Y(1:ex(2)),Z(1:ex(3))
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(inout) :: chi,dxx,dyy,dzz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: trK
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: gxy,gxz,gyz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: Axx,Axy,Axz,Ayy,Ayz,Azz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: Gamx,Gamy,Gamz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(inout) :: Lap, betax, betay, betaz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: dtSfx,  dtSfy,  dtSfz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(in ) :: Sphi,Spi
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(out) :: Sphi_rhs,Spi_rhs
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(inout) :: rho,Sx,Sy,Sz
+  real*8, dimension(ex(1),ex(2),ex(3)),intent(inout) :: Sxx,Sxy,Sxz,Syy,Syz,Szz
+  real*8,intent(in) :: eps
+  integer::gont
+
+  real*8, dimension(ex(1),ex(2),ex(3)) :: gxx,gyy,gzz
+  real*8, dimension(ex(1),ex(2),ex(3)) :: chix,chiy,chiz
+  real*8, dimension(ex(1),ex(2),ex(3)) :: Lapx,Lapy,Lapz
+  real*8, dimension(ex(1),ex(2),ex(3)) :: Kx,Ky,Kz,S
+  real*8, dimension(ex(1),ex(2),ex(3)) :: f,fxx,fxy,fxz,fyy,fyz,fzz
+  real*8, dimension(ex(1),ex(2),ex(3)) :: alpn1,chin1
+  real*8, dimension(ex(1),ex(2),ex(3)) :: gupxx,gupxy,gupxz
+  real*8, dimension(ex(1),ex(2),ex(3)) :: gupyy,gupyz,gupzz
+
+  real*8            :: dX
+  real*8, parameter :: ZEO=0.d0, ONE = 1.D0, TWO = 2.D0, HALF = 0.5D0
+  real*8, parameter :: SYM = 1.D0
+
+  dX = sum(chi)+sum(trK)+sum(dxx)+sum(gxy)+sum(gxz)+sum(dyy)+sum(gyz)+sum(dzz) &
+      +sum(Gamx)+sum(Gamy)+sum(Gamz)                                           &
+      +sum(Lap)+sum(Sphi)+sum(Spi)
+  if(dX.ne.dX) then
+     if(sum(chi).ne.sum(chi))write(*,*)"bssn_escalar_matter: find NaN in chi"
+     if(sum(trK).ne.sum(trK))write(*,*)"bssn_escalar_matter: find NaN in trk"
+     if(sum(dxx).ne.sum(dxx))write(*,*)"bssn_escalar_matter: find NaN in dxx"
+     if(sum(gxy).ne.sum(gxy))write(*,*)"bssn_escalar_matter: find NaN in gxy"
+     if(sum(gxz).ne.sum(gxz))write(*,*)"bssn_escalar_matter: find NaN in gxz"
+     if(sum(dyy).ne.sum(dyy))write(*,*)"bssn_escalar_matter: find NaN in dyy"
+     if(sum(gyz).ne.sum(gyz))write(*,*)"bssn_escalar_matter: find NaN in gyz"
+     if(sum(dzz).ne.sum(dzz))write(*,*)"bssn_escalar_matter: find NaN in dzz"
+     if(sum(Gamx).ne.sum(Gamx))write(*,*)"bssn_escalar_matter: find NaN in Gamx"
+     if(sum(Gamy).ne.sum(Gamy))write(*,*)"bssn_escalar_matter: find NaN in Gamy"
+     if(sum(Gamz).ne.sum(Gamz))write(*,*)"bssn_escalar_matter: find NaN in Gamz"
+     if(sum(Lap).ne.sum(Lap))write(*,*)"bssn_escalar_matter: find NaN in Lap"
+     if(sum(Sphi).ne.sum(Sphi))write(*,*)"bssn_escalar_matter: find NaN in Sphi"
+     if(sum(Spi).ne.sum(Spi))write(*,*)"bssn_escalar_matter: find NaN in Spi"
+     gont = 1
+     return
+  endif
+
+  alpn1 = Lap + ONE
+  chin1 = chi + ONE
+  gxx = dxx + ONE
+  gyy = dyy + ONE
+  gzz = dzz + ONE
+
+  call fderivs(ex,chi,chix,chiy,chiz,X,Y,Z,SYM,SYM,SYM,Symmetry,Lev)
+  call fderivs(ex,Lap,Lapx,Lapy,Lapz,X,Y,Z,SYM,SYM,SYM,Symmetry,Lev)
+
+  gupzz =  gxx * gyy * gzz + gxy * gyz * gxz + gxz * gxy * gyz - &
+           gxz * gyy * gxz - gxy * gxy * gzz - gxx * gyz * gyz
+  gupxx =   ( gyy * gzz - gyz * gyz ) / gupzz
+  gupxy = - ( gxy * gzz - gyz * gxz ) / gupzz
+  gupxz =   ( gxy * gyz - gyy * gxz ) / gupzz
+  gupyy =   ( gxx * gzz - gxz * gxz ) / gupzz
+  gupyz = - ( gxx * gyz - gxy * gxz ) / gupzz
+  gupzz =   ( gxx * gyy - gxy * gxy ) / gupzz
+
+#if 1
+  Sphi_rhs = alpn1 * Spi
+  call fderivs(ex,Sphi,Kx,Ky,Kz,X,Y,Z,SYM,SYM,SYM,Symmetry,Lev)
+  call fdderivs(ex,Sphi,fxx,fxy,fxz,fyy,fyz,fzz,X,Y,Z,SYM,SYM,SYM,Symmetry,Lev)
+  Spi_rhs =   gupxx * fxx + gupyy * fyy + gupzz * fzz +           &
+             ( gupxy * fxy + gupxz * fxz + gupyz * fyz ) * TWO -  &
+             ((Gamx+(gupxx*chix+gupxy*chiy+gupxz*chiz)/TWO/chin1)*Kx  &
+            + (Gamy+(gupxy*chix+gupyy*chiy+gupyz*chiz)/TWO/chin1)*Ky  &
+            + (Gamz+(gupxz*chix+gupyz*chiy+gupzz*chiz)/TWO/chin1)*Kz)
+  Spi_rhs = Spi_rhs*alpn1 +                                      &
+            (gupxx*Lapx*Kx + gupxy*Lapx*Ky + gupxz*Lapx*Kz       &
+            +gupxy*Lapy*Kx + gupyy*Lapy*Ky + gupyz*Lapy*Kz       &
+            +gupxz*Lapz*Kx + gupyz*Lapz*Ky + gupzz*Lapz*Kz)
+
+  call frpotential(ex,Sphi,f,S)
+  Spi_rhs = Spi_rhs*chin1 + alpn1*(trK*Spi - S)
+  rho = chin1*((gupxx * Kx * Kx + gupyy * Ky * Ky + gupzz * Kz * Kz)/TWO + &
+                gupxy * Kx * Ky + gupxz * Kx * Kz + gupyz * Ky * Kz )      &
+        + Spi*Spi/TWO+f
+  Sx = -Spi*Kx
+  Sy = -Spi*Ky
+  Sz = -Spi*Kz
+  f = (rho - Spi*Spi)/chin1
+  Sxx = Kx*Kx-f*gxx
+  Sxy = Kx*Ky-f*gxy
+  Sxz = Kx*Kz-f*gxz
+  Syy = Ky*Ky-f*gyy
+  Syz = Ky*Kz-f*gyz
+  Szz = Kz*Kz-f*gzz
+#else
+  Sphi_rhs = ZEO
+  Spi_rhs = ZEO
+  rho = ZEO
+  Sx = ZEO
+  Sy = ZEO
+  Sz = ZEO
+  Sxx = ZEO
+  Sxy = ZEO
+  Sxz = ZEO
+  Syy = ZEO
+  Syz = ZEO
+  Szz = ZEO
+#endif
+
+  gont = 0
+  return
+end function compute_rhs_bssn_escalar_matter
+
 ! rhs for scalar and GR variables
 ! here we consider vacuum spacetime only
   function compute_rhs_bssn_escalar(ex, T,X, Y, Z,                             &

AMSS_NCKU_source/bssn_class.h

@@ -48,6 +48,7 @@ public:
        double StartTime, TotalTime;
        double AnasTime, DumpTime, d2DumpTime, CheckTime;
        double LastAnas, LastConsOut;
+       bool cuda_level0_constraint_cache_valid;
        int *ConstraintRefreshLevels;
        double Courant;
        double numepss, numepsb, numepsh;
@@ -183,6 +184,9 @@ public:
        virtual void Constraint_Out();
        virtual void Compute_Constraint();
 
+protected:
+       void Initialize_Level_Runtime();
+
 #ifdef With_AHF
 protected:
        MyList<var> *AHList, *AHDList, *GaugeList;

AMSS_NCKU_source/bssn_rhs.h

@@ -6,6 +6,7 @@
 #define f_compute_rhs_bssn compute_rhs_bssn
 #define f_compute_rhs_bssn_ss compute_rhs_bssn_ss
 #define f_compute_rhs_bssn_escalar compute_rhs_bssn_escalar
+#define f_compute_rhs_bssn_escalar_matter compute_rhs_bssn_escalar_matter
 #define f_compute_rhs_bssn_escalar_ss compute_rhs_bssn_escalar_ss
 #define f_compute_rhs_Z4c compute_rhs_z4c
 #define f_compute_rhs_Z4cnot compute_rhs_z4cnot
@@ -16,6 +17,7 @@
 #define f_compute_rhs_bssn COMPUTE_RHS_BSSN
 #define f_compute_rhs_bssn_ss COMPUTE_RHS_BSSN_SS
 #define f_compute_rhs_bssn_escalar COMPUTE_RHS_BSSN_ESCALAR
+#define f_compute_rhs_bssn_escalar_matter COMPUTE_RHS_BSSN_ESCALAR_MATTER
 #define f_compute_rhs_bssn_escalar_ss COMPUTE_RHS_BSSN_ESCALAR_SS
 #define f_compute_rhs_Z4c COMPUTE_RHS_Z4C
 #define f_compute_rhs_Z4cnot COMPUTE_RHS_Z4CNOT
@@ -26,6 +28,7 @@
 #define f_compute_rhs_bssn compute_rhs_bssn_
 #define f_compute_rhs_bssn_ss compute_rhs_bssn_ss_
 #define f_compute_rhs_bssn_escalar compute_rhs_bssn_escalar_
+#define f_compute_rhs_bssn_escalar_matter compute_rhs_bssn_escalar_matter_
 #define f_compute_rhs_bssn_escalar_ss compute_rhs_bssn_escalar_ss_
 #define f_compute_rhs_Z4c compute_rhs_z4c_
 #define f_compute_rhs_Z4cnot compute_rhs_z4cnot_
@@ -96,6 +99,20 @@ extern "C"
                                   int &, int &, double &, int &, int &);
 }
 
+extern "C"
+{
+        int f_compute_rhs_bssn_escalar_matter(int *, double &, double *, double *, double *,                                             // ex,T,X,Y,Z
+                                              double *, double *,                                                                        // chi, trK
+                                              double *, double *, double *, double *, double *, double *,                                // gij
+                                              double *, double *, double *, double *, double *, double *,                                // Aij
+                                              double *, double *, double *,                                                              // Gam
+                                              double *, double *, double *, double *, double *, double *, double *,                      // Gauge
+                                              double *, double *,                                                                        // Sphi, Spi
+                                              double *, double *,                                                                        // Sphi, Spi rhs
+                                              double *, double *, double *, double *, double *, double *, double *, double *, double *, double *, // stress-energy
+                                              int &, int &, double &);
+}
+
 extern "C"
 {
         int f_compute_rhs_bssn_escalar(int *, double &, double *, double *, double *,                                                      // ex,T,X,Y,Z

AMSS_NCKU_source/bssn_rhs_c.C

@@ -1098,12 +1098,12 @@ int f_compute_rhs_bssn(int *ex, double &T,
             betaz_rhs[i] = FF * dtSfz[i];
 
             reta[i] =
-                gupxx[i] * dtSfx_rhs[i] * dtSfx_rhs[i]
+                gupxx[i] * chix[i] * chix[i]
-                + gupyy[i] * dtSfy_rhs[i] * dtSfy_rhs[i]
+                + gupyy[i] * chiy[i] * chiy[i]
-                + gupzz[i] * dtSfz_rhs[i] * dtSfz_rhs[i]
+                + gupzz[i] * chiz[i] * chiz[i]
-                + TWO * ( gupxy[i] * dtSfx_rhs[i] * dtSfy_rhs[i]
+                + TWO * ( gupxy[i] * chix[i] * chiy[i]
-                        + gupxz[i] * dtSfx_rhs[i] * dtSfz_rhs[i]
+                        + gupxz[i] * chix[i] * chiz[i]
-                        + gupyz[i] * dtSfy_rhs[i] * dtSfz_rhs[i] );
+                        + gupyz[i] * chiy[i] * chiz[i] );
 
             #if (GAUGE == 2)
             reta[i] = 1.31 / 2.0 * sqrt( reta[i] / chin1[i] ) / pow( (ONE - sqrt(chin1[i])), 2.0 );
@@ -1116,12 +1116,12 @@ int f_compute_rhs_bssn(int *ex, double &T,
             dtSfz_rhs[i] = Gamz_rhs[i] - reta[i] * dtSfz[i];
             #elif (GAUGE == 4 || GAUGE == 5)
             reta[i] =
-                gupxx[i] * dtSfx_rhs[i] * dtSfx_rhs[i]
+                gupxx[i] * chix[i] * chix[i]
-                + gupyy[i] * dtSfy_rhs[i] * dtSfy_rhs[i]
+                + gupyy[i] * chiy[i] * chiy[i]
-                + gupzz[i] * dtSfz_rhs[i] * dtSfz_rhs[i]
+                + gupzz[i] * chiz[i] * chiz[i]
-                + TWO * ( gupxy[i] * dtSfx_rhs[i] * dtSfy_rhs[i]
+                + TWO * ( gupxy[i] * chix[i] * chiy[i]
-                        + gupxz[i] * dtSfx_rhs[i] * dtSfz_rhs[i]
+                        + gupxz[i] * chix[i] * chiz[i]
-                        + gupyz[i] * dtSfy_rhs[i] * dtSfz_rhs[i] );
+                        + gupyz[i] * chiy[i] * chiz[i] );
 
             #if (GAUGE == 4)
             reta[i] = 1.31 / 2.0 * sqrt( reta[i] / chin1[i] ) / pow( (ONE - sqrt(chin1[i])), 2.0 );

AMSS_NCKU_source/bssn_rhs_cuda.h

@@ -0,0 +1,470 @@
+#ifndef BSSN_RHS_CUDA_H
+#define BSSN_RHS_CUDA_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+enum {
+    BSSN_CUDA_STATE_COUNT = 24,
+    BSSN_CUDA_MATTER_COUNT = 10
+};
+
+int f_compute_rhs_bssn(int *ex, double &T,
+                       double *X, double *Y, double *Z,
+                       double *chi, double *trK,
+                       double *dxx, double *gxy, double *gxz, double *dyy, double *gyz, double *dzz,
+                       double *Axx, double *Axy, double *Axz, double *Ayy, double *Ayz, double *Azz,
+                       double *Gamx, double *Gamy, double *Gamz,
+                       double *Lap, double *betax, double *betay, double *betaz,
+                       double *dtSfx, double *dtSfy, double *dtSfz,
+                       double *chi_rhs, double *trK_rhs,
+                       double *gxx_rhs, double *gxy_rhs, double *gxz_rhs, double *gyy_rhs, double *gyz_rhs, double *gzz_rhs,
+                       double *Axx_rhs, double *Axy_rhs, double *Axz_rhs, double *Ayy_rhs, double *Ayz_rhs, double *Azz_rhs,
+                       double *Gamx_rhs, double *Gamy_rhs, double *Gamz_rhs,
+                       double *Lap_rhs, double *betax_rhs, double *betay_rhs, double *betaz_rhs,
+                       double *dtSfx_rhs, double *dtSfy_rhs, double *dtSfz_rhs,
+                       double *rho, double *Sx, double *Sy, double *Sz,
+                       double *Sxx, double *Sxy, double *Sxz, double *Syy, double *Syz, double *Szz,
+                       double *Gamxxx, double *Gamxxy, double *Gamxxz, double *Gamxyy, double *Gamxyz, double *Gamxzz,
+                       double *Gamyxx, double *Gamyxy, double *Gamyxz, double *Gamyyy, double *Gamyyz, double *Gamyzz,
+                       double *Gamzxx, double *Gamzxy, double *Gamzxz, double *Gamzyy, double *Gamzyz, double *Gamzzz,
+                       double *Rxx, double *Rxy, double *Rxz, double *Ryy, double *Ryz, double *Rzz,
+                       double *ham_Res, double *movx_Res, double *movy_Res, double *movz_Res,
+                       double *Gmx_Res, double *Gmy_Res, double *Gmz_Res,
+                       int &Symmetry, int &Lev, double &eps, int &co);
+
+int bssn_cuda_rk4_substep(void *block_tag,
+                          int *ex, double *X, double *Y, double *Z,
+                          double **state_host_in,
+                          double **state_host_out,
+                          double **matter_host,
+                          const double *propspeed,
+                          const double *soa_flat,
+                          const double *bbox,
+                          double &dT,
+                          double &T,
+                          int &RK4,
+                          int &apply_bam_bc,
+                          int &Symmetry,
+                          int &Lev,
+                          double &eps,
+                          int &co,
+                          int &use_zero_matter,
+                          int &keep_resident_state,
+                          int &apply_enforce_ga,
+                          double &chitiny);
+
+int bssn_cuda_compute_escalar_matter(void *block_tag,
+                                     int *ex, double *X, double *Y, double *Z,
+                                     double **state_host_in,
+                                     double *Sphi_host,
+                                     double *Spi_host,
+                                     double *Sphi_rhs_host,
+                                     double *Spi_rhs_host,
+                                     double a2,
+                                     int &Symmetry,
+                                     int &Lev,
+                                     double &eps,
+                                     int &co,
+                                     int &apply_enforce_ga);
+
+int bssn_cuda_escalar_finalize_scalar_fields(void *block_tag,
+                                             int *ex, double *X, double *Y, double *Z,
+                                             double *Sphi_out_host,
+                                             double *Spi_out_host,
+                                             const double *propspeed,
+                                             const double *soa_flat,
+                                             const double *bbox,
+                                             double &dT,
+                                             int &RK4,
+                                             int &apply_bam_bc,
+                                             int &Symmetry,
+                                             int &Lev,
+                                             double &eps,
+                                             int &precor);
+
+int bssn_cuda_escalar_has_resident_fields(void *block_tag,
+                                          double *Sphi_host,
+                                          double *Spi_host);
+
+int bssn_cuda_escalar_has_any_resident_fields(void *block_tag);
+
+int bssn_cuda_escalar_download_fields_if_present(void *block_tag,
+                                                 int *ex,
+                                                 double *Sphi_host,
+                                                 double *Spi_host);
+
+int bssn_cuda_pack_escalar_batch_to_host_buffer(void *block_tag,
+                                                double **scalar_host_key,
+                                                double *host_buffer,
+                                                int *ex,
+                                                int i0, int j0, int k0,
+                                                int sx, int sy, int sz);
+
+int bssn_cuda_unpack_escalar_batch_from_host_buffer(void *block_tag,
+                                                    double **scalar_host_key,
+                                                    double *host_buffer,
+                                                    int *ex,
+                                                    int i0, int j0, int k0,
+                                                    int sx, int sy, int sz);
+
+int bssn_cuda_pack_escalar_batch_to_device_buffer(void *block_tag,
+                                                  double **scalar_host_key,
+                                                  double *device_buffer,
+                                                  int *ex,
+                                                  int i0, int j0, int k0,
+                                                  int sx, int sy, int sz);
+
+int bssn_cuda_unpack_escalar_batch_from_device_buffer(void *block_tag,
+                                                      double **scalar_host_key,
+                                                      double *device_buffer,
+                                                      int *ex,
+                                                      int i0, int j0, int k0,
+                                                      int sx, int sy, int sz);
+
+int bssn_cuda_restrict_escalar_batch_to_host_buffer(void *block_tag,
+                                                    double **scalar_host_key,
+                                                    double *host_buffer,
+                                                    int *ex,
+                                                    int sx, int sy, int sz,
+                                                    int fi0, int fj0, int fk0,
+                                                    const double *scalar_soa);
+
+int bssn_cuda_prolong_escalar_batch_to_host_buffer(void *block_tag,
+                                                   double **scalar_host_key,
+                                                   double *host_buffer,
+                                                   int *ex,
+                                                   int sx, int sy, int sz,
+                                                   int ii0, int jj0, int kk0,
+                                                   int lbc_i, int lbc_j, int lbc_k,
+                                                   const double *scalar_soa);
+
+int bssn_cuda_restrict_escalar_batch_to_device_buffer(void *block_tag,
+                                                      double **scalar_host_key,
+                                                      double *device_buffer,
+                                                      int *ex,
+                                                      int sx, int sy, int sz,
+                                                      int fi0, int fj0, int fk0,
+                                                      const double *scalar_soa);
+
+int bssn_cuda_prolong_escalar_batch_to_device_buffer(void *block_tag,
+                                                     double **scalar_host_key,
+                                                     double *device_buffer,
+                                                     int *ex,
+                                                     int sx, int sy, int sz,
+                                                     int ii0, int jj0, int kk0,
+                                                     int lbc_i, int lbc_j, int lbc_k,
+                                                     const double *scalar_soa);
+
+int bssn_cuda_prepare_escalar_inter_time_level(void *block_tag,
+                                               int *ex,
+                                               double **src1_host_key,
+                                               double **src2_host_key,
+                                               double **src3_host_key,
+                                               double **dst_host_key,
+                                               int source_count,
+                                               int tindex);
+
+int bssn_cuda_copy_state_region_to_host(void *block_tag,
+                                        int state_index,
+                                        double *host_state,
+                                        int *ex,
+                                        int i0, int j0, int k0,
+                                        int sx, int sy, int sz);
+
+int bssn_cuda_copy_state_region_from_host(void *block_tag,
+                                          int state_index,
+                                          double *host_state,
+                                          int *ex,
+                                          int i0, int j0, int k0,
+                                          int sx, int sy, int sz);
+
+int bssn_cuda_download_resident_state(void *block_tag,
+                                      int *ex,
+                                      double **state_host_out);
+
+int bssn_cuda_download_resident_state_if_present(void *block_tag,
+                                                int *ex,
+                                                double **state_host_out);
+
+int bssn_cuda_resident_state_matches(void *block_tag,
+                                     double **state_host_key);
+
+int bssn_cuda_download_constraint_outputs(int *ex,
+                                          double **constraint_host_out);
+
+int bssn_cuda_pack_state_region_to_host_buffer(void *block_tag,
+                                               int state_index,
+                                               double *host_buffer,
+                                               int *ex,
+                                               int i0, int j0, int k0,
+                                               int sx, int sy, int sz);
+
+int bssn_cuda_interp_state_point3(void *block_tag,
+                                  int *ex,
+                                  int state0,
+                                  int state1,
+                                  int state2,
+                                  double x0,
+                                  double y0,
+                                  double z0,
+                                  double dx,
+                                  double dy,
+                                  double dz,
+                                  double px,
+                                  double py,
+                                  double pz,
+                                  int ordn,
+                                  int symmetry,
+                                  const double *soa3,
+                                  double *out3);
+
+int bssn_cuda_interp_host_two_fields(void *block_tag,
+                                     int *ex,
+                                     double *field0,
+                                     double *field1,
+                                     double x0,
+                                     double y0,
+                                     double z0,
+                                     double dx,
+                                     double dy,
+                                     double dz,
+                                     const double *px,
+                                     const double *py,
+                                     const double *pz,
+                                     int npoints,
+                                     int ordn,
+                                     int symmetry,
+                                     const double *soa6,
+                                     double *out_interleaved);
+
+int bssn_cuda_unpack_state_region_from_host_buffer(void *block_tag,
+                                                   int state_index,
+                                                   double *host_buffer,
+                                                   int *ex,
+                                                   int i0, int j0, int k0,
+                                                   int sx, int sy, int sz);
+
+int bssn_cuda_pack_state_batch_to_host_buffer(void *block_tag,
+                                              int state_count,
+                                              double *host_buffer,
+                                              int *ex,
+                                              int i0, int j0, int k0,
+                                              int sx, int sy, int sz);
+
+int bssn_cuda_pack_state_batch_to_host_buffer_for_host_views(void *block_tag,
+                                                             double **state_host_key,
+                                                             int state_count,
+                                                             double *host_buffer,
+                                                             int *ex,
+                                                             int i0, int j0, int k0,
+                                                             int sx, int sy, int sz);
+
+int bssn_cuda_unpack_state_batch_from_host_buffer(void *block_tag,
+                                                  int state_count,
+                                                  double *host_buffer,
+                                                  int *ex,
+                                                  int i0, int j0, int k0,
+                                                  int sx, int sy, int sz);
+
+int bssn_cuda_unpack_state_batch_from_host_buffer_for_host_views(void *block_tag,
+                                                                 double **state_host_key,
+                                                                 int state_count,
+                                                                 double *host_buffer,
+                                                                 int *ex,
+                                                                 int i0, int j0, int k0,
+                                                                 int sx, int sy, int sz);
+
+int bssn_cuda_restrict_state_batch_to_host_buffer_for_host_views(void *block_tag,
+                                                                 double **state_host_key,
+                                                                 int state_count,
+                                                                 double *host_buffer,
+                                                                 int *ex,
+                                                                 int sx, int sy, int sz,
+                                                                 int fi0, int fj0, int fk0,
+                                                                 const double *state_soa);
+
+int bssn_cuda_restrict_state_batch_to_host_buffer(void *block_tag,
+                                                  int state_count,
+                                                  double *host_buffer,
+                                                  int *ex,
+                                                  int sx, int sy, int sz,
+                                                  int fi0, int fj0, int fk0,
+                                                  const double *state_soa);
+
+int bssn_cuda_prolong_state_batch_to_host_buffer_for_host_views(void *block_tag,
+                                                                double **state_host_key,
+                                                                int state_count,
+                                                                double *host_buffer,
+                                                                int *ex,
+                                                                int sx, int sy, int sz,
+                                                                int ii0, int jj0, int kk0,
+                                                                int lbc_i, int lbc_j, int lbc_k,
+                                                                const double *state_soa);
+
+int bssn_cuda_prolong_state_batch_to_host_buffer(void *block_tag,
+                                                 int state_count,
+                                                 double *host_buffer,
+                                                 int *ex,
+                                                 int sx, int sy, int sz,
+                                                 int ii0, int jj0, int kk0,
+                                                 int lbc_i, int lbc_j, int lbc_k,
+                                                 const double *state_soa);
+
+int bssn_cuda_pack_state_batch_to_device_buffer(void *block_tag,
+                                                 int state_count,
+                                                double *device_buffer,
+                                                int *ex,
+                                                int i0, int j0, int k0,
+                                                int sx, int sy, int sz);
+
+int bssn_cuda_pack_state_batch_to_device_buffer_for_host_views(void *block_tag,
+                                                               double **state_host_key,
+                                                               int state_count,
+                                                               double *device_buffer,
+                                                               int *ex,
+                                                               int i0, int j0, int k0,
+                                                               int sx, int sy, int sz);
+
+int bssn_cuda_unpack_state_batch_from_device_buffer(void *block_tag,
+                                                    int state_count,
+                                                    double *device_buffer,
+                                                    int *ex,
+                                                    int i0, int j0, int k0,
+                                                    int sx, int sy, int sz);
+
+int bssn_cuda_unpack_state_batch_from_device_buffer_for_host_views(void *block_tag,
+                                                                   double **state_host_key,
+                                                                   int state_count,
+                                                                   double *device_buffer,
+                                                                   int *ex,
+                                                                   int i0, int j0, int k0,
+                                                                   int sx, int sy, int sz);
+
+int bssn_cuda_pack_state_segments_to_device_buffer(void *block_tag,
+                                                   int state_count,
+                                                   double *device_buffer,
+                                                   int *ex,
+                                                   int segment_count,
+                                                   const int *segment_meta);
+
+int bssn_cuda_pack_state_segments_to_device_buffer_for_host_views(void *block_tag,
+                                                                  double **state_host_key,
+                                                                  int state_count,
+                                                                  double *device_buffer,
+                                                                  int *ex,
+                                                                  int segment_count,
+                                                                  const int *segment_meta);
+
+int bssn_cuda_unpack_state_segments_from_device_buffer(void *block_tag,
+                                                       int state_count,
+                                                       double *device_buffer,
+                                                       int *ex,
+                                                       int segment_count,
+                                                       const int *segment_meta);
+
+int bssn_cuda_unpack_state_segments_from_device_buffer_for_host_views(void *block_tag,
+                                                                      double **state_host_key,
+                                                                      int state_count,
+                                                                      double *device_buffer,
+                                                                      int *ex,
+                                                                      int segment_count,
+                                                                      const int *segment_meta);
+
+int bssn_cuda_restrict_state_segments_to_device_buffer(void *block_tag,
+                                                       int state_count,
+                                                       double *device_buffer,
+                                                       int *ex,
+                                                       int segment_count,
+                                                       const int *segment_meta);
+
+int bssn_cuda_restrict_state_segments_to_device_buffer_for_host_views(void *block_tag,
+                                                                      double **state_host_key,
+                                                                      int state_count,
+                                                                      double *device_buffer,
+                                                                      int *ex,
+                                                                      int segment_count,
+                                                                      const int *segment_meta,
+                                                                      const double *state_soa);
+
+int bssn_cuda_prolong_state_segments_to_device_buffer(void *block_tag,
+                                                      int state_count,
+                                                      double *device_buffer,
+                                                      int *ex,
+                                                      int segment_count,
+                                                      const int *segment_meta);
+
+int bssn_cuda_prolong_state_segments_to_device_buffer_for_host_views(void *block_tag,
+                                                                     double **state_host_key,
+                                                                     int state_count,
+                                                                     double *device_buffer,
+                                                                     int *ex,
+                                                                     int segment_count,
+                                                                     const int *segment_meta,
+                                                                     const double *state_soa);
+
+int bssn_cuda_restrict_state_batch_to_device_buffer(void *block_tag,
+                                                    int state_count,
+                                                    double *device_buffer,
+                                                    int *ex,
+                                                    int sx, int sy, int sz,
+                                                    int fi0, int fj0, int fk0);
+
+int bssn_cuda_restrict_state_batch_to_device_buffer_for_host_views(void *block_tag,
+                                                                   double **state_host_key,
+                                                                   int state_count,
+                                                                   double *device_buffer,
+                                                                   int *ex,
+                                                                   int sx, int sy, int sz,
+                                                                   int fi0, int fj0, int fk0,
+                                                                   const double *state_soa);
+
+int bssn_cuda_prolong_state_batch_to_device_buffer(void *block_tag,
+                                                   int state_count,
+                                                   double *device_buffer,
+                                                   int *ex,
+                                                   int sx, int sy, int sz,
+                                                   int ii0, int jj0, int kk0,
+                                                   int lbc_i, int lbc_j, int lbc_k);
+
+int bssn_cuda_prolong_state_batch_to_device_buffer_for_host_views(void *block_tag,
+                                                                  double **state_host_key,
+                                                                  int state_count,
+                                                                  double *device_buffer,
+                                                                  int *ex,
+                                                                  int sx, int sy, int sz,
+                                                                  int ii0, int jj0, int kk0,
+                                                                  int lbc_i, int lbc_j, int lbc_k,
+                                                                  const double *state_soa);
+
+int bssn_cuda_download_state_subset(void *block_tag,
+                                    int *ex,
+                                    int subset_count,
+                                    const int *state_indices,
+                                    double **state_host_out);
+
+int bssn_cuda_upload_state_subset(void *block_tag,
+                                  int *ex,
+                                  int subset_count,
+                                  const int *state_indices,
+                                  double **state_host_in);
+
+int bssn_cuda_prepare_inter_time_level(void *block_tag,
+                                       int *ex,
+                                       double **src1_host_key,
+                                       double **src2_host_key,
+                                       double **src3_host_key,
+                                       double **dst_host_key,
+                                       int source_count,
+                                       int tindex);
+
+int bssn_cuda_has_resident_state(void *block_tag);
+
+void bssn_cuda_release_step_ctx(void *block_tag);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif