ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Avoid (harmless) errors when shutting down the GPU.

Browse Source
This commit is contained in:
Axel Kohlmeyer
2024-01-18 22:42:32 -05:00
parent 54794a45de
commit 991663feff
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/gpu/geryon/nvd_device.h
View File

@ -138,7 +138,7 @@ class UCL_Device {
/** \note You cannot delete the default stream **/ /** \note You cannot delete the default stream **/
inline void pop_command_queue() { inline void pop_command_queue() {
if (_cq.size()<2) return; if (_cq.size()<2) return;
CU_SAFE_CALL_NS(cuStreamDestroy(_cq.back())); cuStreamDestroy(_cq.back());
_cq.pop_back(); _cq.pop_back();
} }
@ -426,8 +426,8 @@ void UCL_Device::clear() {
if (_device>-1) { if (_device>-1) {
for (int i=1; i<num_queues(); i++) pop_command_queue(); for (int i=1; i<num_queues(); i++) pop_command_queue();
#if GERYON_NVD_PRIMARY_CONTEXT #if GERYON_NVD_PRIMARY_CONTEXT
CU_SAFE_CALL_NS(cuCtxSetCurrent(_old_context)); cuCtxSetCurrent(_old_context);
CU_SAFE_CALL_NS(cuDevicePrimaryCtxRelease(_cu_device)); cuDevicePrimaryCtxRelease(_cu_device);
#else #else
cuCtxDestroy(_context); cuCtxDestroy(_context);
#endif #endif