Skip to content

GitLab

Commit e0e5a84c authored by rusty1s's avatar rusty1s
Browse files

fix equal outcome in radius for CPU/GPU

parent 2582d47b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/cpu/radius_cpu.cpp
View file @ e0e5a84c
...@@ -52,7 +52,7 @@ torch::Tensor radius_cpu(torch::Tensor x, torch::Tensor y, ...@@ -52,7 +52,7 @@ torch::Tensor radius_cpu(torch::Tensor x, torch::Tensor y,
for (int64_t i = 0; i < y.size(0); i++) { for (int64_t i = 0; i < y.size(0); i++) {
std::vector<std::pair<size_t, scalar_t>> ret_matches; std::vector<std::pair<size_t, scalar_t>> ret_matches;
size_t num_matches = mat_index.index->radiusSearch( size_t num_matches = mat_index.index->radiusSearch(
y_data + i * y.size(1), r * r + 0.00001, ret_matches, params); y_data + i * y.size(1), r * r, ret_matches, params);
for (size_t j = 0; j < std::min(num_matches, (size_t)max_num_neighbors); for (size_t j = 0; j < std::min(num_matches, (size_t)max_num_neighbors);
j++) { j++) {
...@@ -86,7 +86,7 @@ torch::Tensor radius_cpu(torch::Tensor x, torch::Tensor y, ...@@ -86,7 +86,7 @@ torch::Tensor radius_cpu(torch::Tensor x, torch::Tensor y,
for (int64_t i = y_start; i < y_end; i++) { for (int64_t i = y_start; i < y_end; i++) {
std::vector<std::pair<size_t, scalar_t>> ret_matches; std::vector<std::pair<size_t, scalar_t>> ret_matches;
size_t num_matches = mat_index.index->radiusSearch( size_t num_matches = mat_index.index->radiusSearch(
y_data + i * y.size(1), r * r + 0.00001, ret_matches, params); y_data + i * y.size(1), r * r, ret_matches, params);
for (size_t j = 0; for (size_t j = 0;
j < std::min(num_matches, (size_t)max_num_neighbors); j++) { j < std::min(num_matches, (size_t)max_num_neighbors); j++) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment