fix KNN CUDA bug (PyG #4906)

113e8a6a · rusty1s · a2773c21 · 113e8a6a
Commit 113e8a6a authored Jul 07, 2022 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/cuda/knn_cuda.cu csrc/cuda/knn_cuda.cu +1 -1

No files found.
--- a/csrc/cuda/knn_cuda.cu
+++ b/csrc/cuda/knn_cuda.cu
@@ -45,7 +45,7 @@ knn_kernel(const scalar_t *__restrict__ x, const scalar_t *__restrict__ y,
  int64_t best_idx[100];

  for (int e = 0; e < k; e++) {
-    best_dist[e] = 5e4;
+    best_dist[e] = 1e10;
    best_idx[e] = -1;
  }