Batch GPU stage downloads

2026-04-12 21:06:41 +08:00
parent 87c581ea7c
commit 7064ebd5b4
3 changed files with 94 additions and 10 deletions
--- a/AMSS_NCKU_source/bssn_gpu.h
+++ b/AMSS_NCKU_source/bssn_gpu.h
@@ -99,6 +99,7 @@ int bssn_gpu_stage_upload_buffer_to_region(const double *host_src_ptr,
                                           const double *full_uub,
                                           const int *region_shape,
                                           const double *region_llb);
+int bssn_gpu_download_buffer_batch(const int *ex, double **host_ptrs, int num_buffers);

 /** Init GPU side data in GPUMeta. */
 // void init_fluid_meta_gpu(GPUMeta *gpu_meta);