Spaces:
Sleeping
Sleeping
Diego Devesa
commited on
Commit
·
106cf6f
1
Parent(s):
295521c
cuda : clear error after changing peer access (llama/10153)
Browse files- 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 |
}
|