Checkpoint Z4C CUDA throttling progress

2026-05-02 10:04:23 +08:00
parent fcd98649f6
commit ba61702fc0
2 changed files with 34 additions and 1 deletions
--- a/AMSS_NCKU_source/Parallel.C
+++ b/AMSS_NCKU_source/Parallel.C
@@ -5319,6 +5319,10 @@ int Parallel::data_packer(double *data, MyList<Parallel::gridseg> *src, MyList<P
                    MPI_Abort(MPI_COMM_WORLD, 1);
                  }
                }
+                else if (z4c_cuda_has_resident_state(dst->data->Bg) != 0)
+                {
+                  z4c_cuda_release_step_ctx(dst->data->Bg);
+                }
              }
              else
 #endif