"src/vscode:/vscode.git/clone" did not exist on "0feb21a18c44cfbf76a916afead986f04b339292"
Commit d4fe021d authored by Alexander Liao's avatar Alexander Liao
Browse files

attempting to fix win/os build error

parent d7f704c5
......@@ -63,8 +63,8 @@ torch::Tensor batch_radius_cpu(torch::Tensor query,
float radius, int max_num) {
torch::Tensor out;
auto data_qb = query_batch.data_ptr<long>();
auto data_sb = support_batch.data_ptr<long>();
auto data_qb = query_batch.data_ptr<int64_t>();
auto data_sb = support_batch.data_ptr<int64_t>();
std::vector<long> query_batch_stl = std::vector<long>(data_qb, data_qb+query_batch.size(0));
std::vector<long> size_query_batch_stl;
get_size_batch(query_batch_stl, size_query_batch_stl);
......
......@@ -7,17 +7,11 @@ template<typename scalar_t>
int nanoflann_neighbors(vector<scalar_t>& queries, vector<scalar_t>& supports,
vector<long>& neighbors_indices, float radius, int dim, int max_num){
// Initiate variables
// ******************
const scalar_t search_radius = static_cast<scalar_t>(radius*radius);
// Counting vector
size_t max_count = 1;
// Nanoflann related variables
// ***************************
// CLoud variable
PointCloud<scalar_t> pcd;
pcd.set(supports, dim);
......@@ -54,32 +48,15 @@ int nanoflann_neighbors(vector<scalar_t>& queries, vector<scalar_t>& supports,
scalar_t* query_pt = new scalar_t[dim];
std::copy(p0.begin(), p0.end(), query_pt);
//for(int i=0; i < p0.size(); i++)
//std::cout << query_pt[i] << '\n';
list_matches[i0].reserve(max_count);
std::vector<std::pair<size_t, scalar_t> > ret_matches;
const size_t nMatches = index->radiusSearch(query_pt, search_radius+eps, ret_matches, search_params);
//cout << "radiusSearch(): radius=" << search_radius << " -> " << nMatches << " matches\n";
//for (size_t i = 0; i < nMatches; i++)
// cout << "idx["<< i << "]=" << ret_matches[i].first << " dist["<< i << "]=" << ret_matches[i].second << endl;
//cout << "\n";
list_matches[i0] = ret_matches;
if(max_count < nMatches) max_count = nMatches;
i0++;
// Get worst (furthest) point, without sorting:
// cout << "\n neighbors: " << nMatches << "\n";
// Get worst (furthest) point, without sorting:
// for(int i=0; i < ret_matches.size(); i++)
// std::cout << ret_matches.at(i) << '\n';
}
// Reserve the memory
if(max_num > 0) {
......
......@@ -4,6 +4,7 @@
#include "nanoflann.hpp"
#include <set>
#include <cstdint>
#include <thread>
using namespace std;
......
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