Răsfoiți Sursa

cuda : clear error after changing peer access (#10153)

Diego Devesa 1 an în urmă
părinte
comite
ea02c753eb
1 a modificat fișierele cu 6 adăugiri și 0 ștergeri
  1. 6 0
      ggml/src/ggml-cuda.cu

+ 6 - 0
ggml/src/ggml-cuda.cu

@@ -1297,11 +1297,17 @@ static void ggml_cuda_set_peer_access(const int n_tokens, int main_device) {
                     cudaError_t err = cudaDeviceEnablePeerAccess(id_other, 0);
                     if (err != cudaErrorPeerAccessAlreadyEnabled) {
                         CUDA_CHECK(err);
+                    } else {
+                        // reset the error
+                        cudaGetLastError();
                     }
                 } else {
                     cudaError_t err = cudaDeviceDisablePeerAccess(id_other);
                     if (err != cudaErrorPeerAccessNotEnabled) {
                         CUDA_CHECK(err);
+                    } else {
+                        // reset the error
+                        cudaGetLastError();
                     }
                 }
             }