Removed bad conversions

Author: Bernhard Kerbl

cuda_rasterizer/rasterizer.h

@@ -18,9 +18,9 @@ namespace CudaRasterizer
 			bool* present);
 
 		static int forward(
-			std::function<char* (int)> geometryBuffer,
-			std::function<char* (int)> binningBuffer,
-			std::function<char* (int)> imageBuffer,
+			std::function<char* (size_t)> geometryBuffer,
+			std::function<char* (size_t)> binningBuffer,
+			std::function<char* (size_t)> imageBuffer,
 			const int P, int D, int M,
 			const float* background,
 			const int width, int height,

cuda_rasterizer/rasterizer_impl.cu

@@ -141,7 +141,7 @@ void CudaRasterizer::Rasterizer::markVisible(
 		present);
 }
 
-CudaRasterizer::GeometryState CudaRasterizer::GeometryState::fromChunk(char*& chunk, int P)
+CudaRasterizer::GeometryState CudaRasterizer::GeometryState::fromChunk(char*& chunk, size_t P)
 {
 	GeometryState geom;
 	obtain(chunk, geom.depths, P, 128);
@@ -158,7 +158,7 @@ CudaRasterizer::GeometryState CudaRasterizer::GeometryState::fromChunk(char*& ch
 	return geom;
 }
 
-CudaRasterizer::ImageState CudaRasterizer::ImageState::fromChunk(char*& chunk, int N)
+CudaRasterizer::ImageState CudaRasterizer::ImageState::fromChunk(char*& chunk, size_t N)
 {
 	ImageState img;
 	obtain(chunk, img.accum_alpha, N, 128);
@@ -167,7 +167,7 @@ CudaRasterizer::ImageState CudaRasterizer::ImageState::fromChunk(char*& chunk, i
 	return img;
 }
 
-CudaRasterizer::BinningState CudaRasterizer::BinningState::fromChunk(char*& chunk, int P)
+CudaRasterizer::BinningState CudaRasterizer::BinningState::fromChunk(char*& chunk, size_t P)
 {
 	BinningState binning;
 	obtain(chunk, binning.point_list, P, 128);
@@ -185,9 +185,9 @@ CudaRasterizer::BinningState CudaRasterizer::BinningState::fromChunk(char*& chun
 // Forward rendering procedure for differentiable rasterization
 // of Gaussians.
 int CudaRasterizer::Rasterizer::forward(
-	std::function<char*(int)> geometryBuffer,
-	std::function<char* (int)> binningBuffer,
-	std::function<char* (int)> imageBuffer,
+	std::function<char* (size_t)> geometryBuffer,
+	std::function<char* (size_t)> binningBuffer,
+	std::function<char* (size_t)> imageBuffer,
 	const int P, int D, int M,
 	const float* background,
 	const int width, int height,
@@ -210,7 +210,7 @@ int CudaRasterizer::Rasterizer::forward(
 	const float focal_y = height / (2.0f * tan_fovy);
 	const float focal_x = width / (2.0f * tan_fovx);
 
-	int chunk_size = required<GeometryState>(P);
+	size_t chunk_size = required<GeometryState>(P);
 	char* chunkptr = geometryBuffer(chunk_size);
 	GeometryState geomState = GeometryState::fromChunk(chunkptr, P);
 

cuda_rasterizer/rasterizer_impl.h

@@ -29,7 +29,7 @@ namespace CudaRasterizer
 		uint32_t* point_offsets;
 		uint32_t* tiles_touched;
 
-		static GeometryState fromChunk(char*& chunk, int P);
+		static GeometryState fromChunk(char*& chunk, size_t P);
 	};
 
 	struct ImageState
@@ -38,7 +38,7 @@ namespace CudaRasterizer
 		uint32_t* n_contrib;
 		float* accum_alpha;
 
-		static ImageState fromChunk(char*& chunk, int N);
+		static ImageState fromChunk(char*& chunk, size_t N);
 	};
 
 	struct BinningState
@@ -50,14 +50,14 @@ namespace CudaRasterizer
 		uint32_t* point_list;
 		char* list_sorting_space;
 
-		static BinningState fromChunk(char*& chunk, int P);
+		static BinningState fromChunk(char*& chunk, size_t P);
 	};
 
 	template<typename T> 
-	int required(int P)
+	size_t required(size_t P)
 	{
 		char* size = nullptr;
 		T::fromChunk(size, P);
-		return ((int)size) + 128;
+		return ((size_t)size) + 128;
 	}
 };

rasterize_points.cu

@@ -15,9 +15,9 @@
 #include <string>
 #include <functional>
 
-std::function<char*(int N)> resizeFunctional(torch::Tensor& t) {
-    auto lambda = [&t](int N) {
-        t.resize_({N});
+std::function<char*(size_t N)> resizeFunctional(torch::Tensor& t) {
+    auto lambda = [&t](size_t N) {
+        t.resize_({(long long)N});
 		return reinterpret_cast<char*>(t.contiguous().data_ptr());
     };
     return lambda;
@@ -64,9 +64,9 @@ RasterizeGaussiansCUDA(
   torch::Tensor geomBuffer = torch::empty({0}, options.device(device));
   torch::Tensor binningBuffer = torch::empty({0}, options.device(device));
   torch::Tensor imgBuffer = torch::empty({0}, options.device(device));
-  std::function<char*(int)> geomFunc = resizeFunctional(geomBuffer);
-  std::function<char*(int)> binningFunc = resizeFunctional(binningBuffer);
-  std::function<char*(int)> imgFunc = resizeFunctional(imgBuffer);
+  std::function<char*(size_t)> geomFunc = resizeFunctional(geomBuffer);
+  std::function<char*(size_t)> binningFunc = resizeFunctional(binningBuffer);
+  std::function<char*(size_t)> imgFunc = resizeFunctional(imgBuffer);
 
   int rendered = 0;
   if(P != 0)