fix KNN CUDA bug (PyG #4906)

rusty1s · rusty1s · commit 113e8a6abbe3 · 2022-07-07T09:09:02.000Z
diff --git 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;
   }
 

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ knn_kernel(const scalar_t __restrict__ x, const scalar_t __restrict__ y,`
`45`	`45`	`int64_t best_idx[100];`
`46`	`46`
`47`	`47`	`for (int e = 0; e < k; e++) {`
`48`		`- best_dist[e] = 5e4;`
	`48`	`+ best_dist[e] = 1e10;`
`49`	`49`	`best_idx[e] = -1;`
`50`	`50`	`}`
`51`	`51`