diff options
Diffstat (limited to 'nerv/lib/matrix/cumatrix.c')
| -rw-r--r-- | nerv/lib/matrix/cumatrix.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/lib/matrix/cumatrix.c b/nerv/lib/matrix/cumatrix.c index ff2ea22..537fabb 100644 --- a/nerv/lib/matrix/cumatrix.c +++ b/nerv/lib/matrix/cumatrix.c @@ -78,10 +78,10 @@ void nerv_cuda_context_destroy(CuContext *context, Status *status) { void nerv_cuda_context_select_gpu(CuContext *context, int dev, Status *status) { + CUDA_SAFE_SYNC_CALL(cudaSetDevice(dev), status); free_cuda_handles(context, status); if (status->err_code != NERV_NORMAL) return; - CUDA_SAFE_SYNC_CALL(cudaSetDevice(dev), status); new_cuda_handles(context, status); if (status->err_code != NERV_NORMAL) return; |