attempting to fix win/os build error

Author: liaopeiyuan

csrc/cpu/radius_cpu.cpp

@@ -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);

csrc/cpu/utils/neighbors.cpp

@@ -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) {

csrc/cpu/utils/neighbors.h

@@ -4,6 +4,7 @@
 #include "nanoflann.hpp"
 #include <set>
 #include <cstdint>
+#include <thread>
 
 using namespace std;