fix installation bug + cpu/cuda unequal output bug

rusty1s · rusty1s · commit f99e7ffbd1ad · 2021-02-26T10:25:45.000+01:00
diff --git 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++;
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ __global__ void radius_kernel(const scalar_t x, const scalar_t y,`
`31`	`31`	`}`
`32`	`32`	`dist = sqrt(dist);`
`33`	`33`
`34`		`- if (dist <= radius) {`
	`34`	`+ if (dist < radius) {`
`35`	`35`	`row[n_y * max_num_neighbors + count] = n_y;`
`36`	`36`	`col[n_y * max_num_neighbors + count] = n_x;`
`37`	`37`	`count++;`