Diego Devesa commited on
Commit
106cf6f
·
1 Parent(s): 295521c

cuda : clear error after changing peer access (llama/10153)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-cuda.cu +6 -0
ggml/src/ggml-cuda.cu CHANGED
@@ -1297,11 +1297,17 @@ static void ggml_cuda_set_peer_access(const int n_tokens, int main_device) {
1297
  cudaError_t err = cudaDeviceEnablePeerAccess(id_other, 0);
1298
  if (err != cudaErrorPeerAccessAlreadyEnabled) {
1299
  CUDA_CHECK(err);
 
 
 
1300
  }
1301
  } else {
1302
  cudaError_t err = cudaDeviceDisablePeerAccess(id_other);
1303
  if (err != cudaErrorPeerAccessNotEnabled) {
1304
  CUDA_CHECK(err);
 
 
 
1305
  }
1306
  }
1307
  }
 
1297
  cudaError_t err = cudaDeviceEnablePeerAccess(id_other, 0);
1298
  if (err != cudaErrorPeerAccessAlreadyEnabled) {
1299
  CUDA_CHECK(err);
1300
+ } else {
1301
+ // reset the error
1302
+ cudaGetLastError();
1303
  }
1304
  } else {
1305
  cudaError_t err = cudaDeviceDisablePeerAccess(id_other);
1306
  if (err != cudaErrorPeerAccessNotEnabled) {
1307
  CUDA_CHECK(err);
1308
+ } else {
1309
+ // reset the error
1310
+ cudaGetLastError();
1311
  }
1312
  }
1313
  }