fix installation bug + cpu/cuda unequal output bug

f99e7ffb · rusty1s · 234a6ba3 · f99e7ffb · f99e7ffb
Commit f99e7ffb authored Feb 26, 2021 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

setup.py setup.py +1 -1

No files found.
--- a/csrc/cuda/radius_cuda.cu
+++ b/csrc/cuda/radius_cuda.cu
@@ -31,7 +31,7 @@ __global__ void radius_kernel(const scalar_t *x, const scalar_t *y,
      }
      dist = sqrt(dist);
-      if (dist <= radius) {
+      if (dist < radius) {
        row[n_y * max_num_neighbors + count] = n_y;
        col[n_y * max_num_neighbors + count] = n_x;
        count++;

--- a/setup.py
+++ b/setup.py
@@ -58,7 +58,7 @@ def get_extensions():
            sources += [path]
        path = osp.join(extensions_dir, 'cuda', f'{name}_cuda.cu')
-        if WITH_CUDA and osp.exists(path):
+        if suffix == 'cuda' and osp.exists(path):
            sources += [path]
        Extension = CppExtension if suffix == 'cpu' else CUDAExtension