64-BitBrainstorm_2026/AMSS-NCKU
8
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Register GPU transfer buffers as pinned host memory

Browse Source
This commit is contained in:
CGH0S7
2026-04-09 18:36:10 +08:00
parent 5bc67ded06
commit b0dd069a2b
3 changed files with 128 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
AMSS_NCKU_source/bssn_gpu.h
View File

@@ -69,6 +69,7 @@ int bssn_gpu_bind_process_device(int mpi_rank);
void bssn_gpu_clear_cached_device_buffers();
const double *bssn_gpu_find_device_buffer(const double *host_ptr);
void bssn_gpu_register_device_buffer(const double *host_ptr, const double *device_ptr);
void bssn_gpu_prepare_host_buffer(const double *host_ptr, int count);
int bssn_gpu_stage_upload_buffer(const double *host_ptr, int count);
int bssn_gpu_stage_zero_buffer(const double *host_ptr, int count);
int bssn_gpu_stage_upload_region(const double *host_ptr,