api_cuda

2

6

7namespace rocky::cuda

8{

9

10inline void sync_devices(IRockyModel &model)

11{

12 // Sync all GPUs

13 for (int i = 0; i < model.get_number_of_devices(); i++)

14 {

15 CUDA_SET_DEVICE(model.get_device_id(i));

16 CUDA_ERROR_CHECK(cudaDeviceSynchronize());

17 }

18}

19

20}