Optimize GPU RK4 stage sync path

This commit is contained in:
2026-04-12 18:36:05 +08:00
parent d9287ea530
commit c8977d8356
4 changed files with 698 additions and 33 deletions

View File

@@ -23,7 +23,10 @@ using namespace std;
#include <mpi.h>
#include "misc.h"
#include "macrodef.h"
#include "macrodef.h"
#ifdef USE_GPU
extern void bssn_cuda_dump_stage_profile();
#endif
#ifndef ABEtype
#error "not define ABEtype"
@@ -469,10 +472,13 @@ int main(int argc, char *argv[])
cout << endl;
}
ADM->Evolve(Steps);
if (myrank == 0)
{
ADM->Evolve(Steps);
#ifdef USE_GPU
bssn_cuda_dump_stage_profile();
#endif
if (myrank == 0)
{
cout << endl;
cout << " Total Evolve Time: " << MPI_Wtime() - End_clock << " seconds!" << endl;
cout << " Total Running Time: " << MPI_Wtime() - Begin_clock << " seconds!" << endl;