typo

6f0a5a4d · rusty1s · 685b3770 · 6f0a5a4d
Commit 6f0a5a4d authored Nov 13, 2018 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

cuda/fps_kernel.cu cuda/fps_kernel.cu +2 -3

No files found.
--- a/cuda/fps_kernel.cu
+++ b/cuda/fps_kernel.cu
@@ -15,9 +15,8 @@ template <typename scalar_t> struct Dist<scalar_t, 1> {
          scalar_t *__restrict__ tmp_dist, size_t dim) {
    for (ptrdiff_t n = start_idx + idx; n < end_idx; n += THREADS) {
-      scalar_t a = x[old * 3 + 0] - x[n * 3 + 0];
+      scalar_t d = x[old * 3 + 0] - x[n * 3 + 0];
-      scalar_t d = a * a;
+      dist[n] = min(dist[n], d * d);
-      dist[n] = min(dist[n], d);
      if (dist[n] > *best) {
        *best = dist[n];
        *best_idx = n;