Skip to content

Commit abcd5b1

Browse files
author
Y
committed
Added cudaFree inside safeCall to handle invalid mem access
1 parent 958a718 commit abcd5b1

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

QuickSilver/CUDA/src/main.cc

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,7 @@ void copyMonteCarloHost_part(MonteCarlo_d* mc_d, MonteCarlo* mc)
334334
safeCall(cudaMemcpy(mc->time_info, tmp, sizeof(MC_Time_Info), cudaMemcpyDeviceToHost));
335335
safeCall(cudaDeviceSynchronize());
336336
//sycl::free(tmp,sycl_device_queue);
337-
cudaFree(tmp);
337+
safeCall(cudaFree(tmp));
338338

339339
ParticleVaultContainer_d* tmp_p;
340340
//sycl_device_queue.memcpy(&tmp_p, &(mc_d->_particleVaultContainer_d), sizeof(ParticleVaultContainer_d*)).wait();
@@ -362,15 +362,15 @@ void copyMonteCarloHost_part(MonteCarlo_d* mc_d, MonteCarlo* mc)
362362
ParticleVault_d *tmp_0;
363363
//sycl_device_queue.memcpy(&tmp_0, &(tmp_ev[i]), sizeof(ParticleVault_d *)).wait();
364364
safeCall(cudaMemcpy(&tmp_0, &(tmp_ev[i]), sizeof(ParticleVault_d *), cudaMemcpyDeviceToHost));
365-
safeCall(cudaDeviceSynchronize());
365+
safeCall(cudaDeviceSynchronize());
366366
copyParticleVault_d2h(mc->_particleVaultContainer->_extraVault[i], tmp_0);
367367
//sycl::free(tmp_0, sycl_device_queue);
368-
cudaFree(tmp_0);
368+
safeCall(cudaFree(tmp_0));
369369
}
370370
//sycl::free(tmp_ev, sycl_device_queue);
371-
cudaFree(tmp_ev);
371+
safeCall(cudaFree(tmp_ev));
372372
//sycl::free(tmp_p, sycl_device_queue);
373-
cudaFree(tmp_p);
373+
safeCall(cudaFree(tmp_p));
374374
}
375375

376376
void copyMonteCarloHost_last(MonteCarlo_d* mc_d, MonteCarlo* mc)

0 commit comments

Comments
 (0)