diff options
author | Determinant <ted.sybil@gmail.com> | 2016-03-02 14:59:59 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-03-02 14:59:59 +0800 |
commit | 75108e2e000f4382129d453c0fa9073b14d32f97 (patch) | |
tree | 9b952b8a39affdc7725354ec41940ce5af9d4629 /nerv/lib | |
parent | 42608086c666e23ceac0e9046836a8fefe00e5b2 (diff) |
add options to main entry `nerv`; fix a bug in handling `cudaSetDevice` error
Diffstat (limited to 'nerv/lib')
-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; |