cwpearson/astaroth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added some error checking for the case where user supplies an incomplete meshinfo to acDeviceLoadMeshInfo

Browse Source
This commit is contained in:
jpekkila
2019-08-19 15:17:51 +03:00
parent 598799d7c3
commit 41805dcb68
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/core/device.cu
View File

@@ -370,6 +370,13 @@ AcResult
acDeviceLoadMeshInfo(const Device device, const Stream stream, const AcMeshInfo device_config) acDeviceLoadMeshInfo(const Device device, const Stream stream, const AcMeshInfo device_config)
{ {
cudaSetDevice(device->id); cudaSetDevice(device->id);
ERRCHK_ALWAYS(device_config.int_params[AC_nx] == device->local_config.int_params[AC_nx]);
ERRCHK_ALWAYS(device_config.int_params[AC_ny] == device->local_config.int_params[AC_ny]);
ERRCHK_ALWAYS(device_config.int_params[AC_nz] == device->local_config.int_params[AC_nz]);
ERRCHK_ALWAYS(device_config.int_params[AC_multigpu_offset] ==
device->local_config.int_params[AC_multigpu_offset]);
ERRCHK_CUDA_ALWAYS(cudaMemcpyToSymbolAsync(d_mesh_info, &device_config, sizeof(device_config), ERRCHK_CUDA_ALWAYS(cudaMemcpyToSymbolAsync(d_mesh_info, &device_config, sizeof(device_config),
0, cudaMemcpyHostToDevice, device->streams[stream])); 0, cudaMemcpyHostToDevice, device->streams[stream]));
return AC_SUCCESS; return AC_SUCCESS;