From 2d20cc464f919a0731b0bd819e9c4d396a34749c Mon Sep 17 00:00:00 2001
From: zjjMaiMai <13913992+zjjMaiMai@users.noreply.github.com>
Date: Sun, 24 Jul 2022 17:33:17 +0800
Subject: [PATCH] [fix] vren build error on windows cause long

---
 models/csrc/intersection.cu    | 8 ++++----
 models/csrc/raymarching.cu     | 4 ++--
 models/csrc/volumerendering.cu | 4 ++--
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/models/csrc/intersection.cu b/models/csrc/intersection.cu
index 172b45fad..635968e47 100644
--- a/models/csrc/intersection.cu
+++ b/models/csrc/intersection.cu
@@ -30,7 +30,7 @@ __global__ void ray_aabb_intersect_kernel(
     const int max_hits,
     int* hit_cnt,
     torch::PackedTensorAccessor32<float, 3, torch::RestrictPtrTraits> hits_t,
-    torch::PackedTensorAccessor64<long, 2, torch::RestrictPtrTraits> hits_voxel_idx
+    torch::PackedTensorAccessor64<int64_t, 2, torch::RestrictPtrTraits> hits_voxel_idx
 ){
     const int r = blockIdx.x * blockDim.x + threadIdx.x;
     const int v = blockIdx.y * blockDim.y + threadIdx.y;
@@ -87,7 +87,7 @@ std::vector<torch::Tensor> ray_aabb_intersect_cu(
             max_hits,
             hit_cnt.data_ptr<int>(),
             hits_t.packed_accessor32<float, 3, torch::RestrictPtrTraits>(),
-            hits_voxel_idx.packed_accessor64<long, 2, torch::RestrictPtrTraits>()
+            hits_voxel_idx.packed_accessor64<int64_t, 2, torch::RestrictPtrTraits>()
         );
     }));
 
@@ -129,7 +129,7 @@ __global__ void ray_sphere_intersect_kernel(
     const int max_hits,
     int* hit_cnt,
     torch::PackedTensorAccessor32<float, 3, torch::RestrictPtrTraits> hits_t,
-    torch::PackedTensorAccessor64<long, 2, torch::RestrictPtrTraits> hits_sphere_idx
+    torch::PackedTensorAccessor64<int64_t, 2, torch::RestrictPtrTraits> hits_sphere_idx
 ){
     const int r = blockIdx.x * blockDim.x + threadIdx.x;
     const int s = blockIdx.y * blockDim.y + threadIdx.y;
@@ -184,7 +184,7 @@ std::vector<torch::Tensor> ray_sphere_intersect_cu(
             max_hits,
             hit_cnt.data_ptr<int>(),
             hits_t.packed_accessor32<float, 3, torch::RestrictPtrTraits>(),
-            hits_sphere_idx.packed_accessor64<long, 2, torch::RestrictPtrTraits>()
+            hits_sphere_idx.packed_accessor64<int64_t, 2, torch::RestrictPtrTraits>()
         );
     }));
 
diff --git a/models/csrc/raymarching.cu b/models/csrc/raymarching.cu
index 9e2629956..6d42e653a 100644
--- a/models/csrc/raymarching.cu
+++ b/models/csrc/raymarching.cu
@@ -336,7 +336,7 @@ __global__ void raymarching_test_kernel(
     const torch::PackedTensorAccessor32<float, 2, torch::RestrictPtrTraits> rays_o,
     const torch::PackedTensorAccessor32<float, 2, torch::RestrictPtrTraits> rays_d,
     torch::PackedTensorAccessor32<float, 2, torch::RestrictPtrTraits> hits_t,
-    const torch::PackedTensorAccessor64<long, 1, torch::RestrictPtrTraits> alive_indices,
+    const torch::PackedTensorAccessor64<int64_t, 1, torch::RestrictPtrTraits> alive_indices,
     const uint8_t* __restrict__ density_bitfield,
     const int cascades,
     const int grid_size,
@@ -434,7 +434,7 @@ std::vector<torch::Tensor> raymarching_test_cu(
             rays_o.packed_accessor32<float, 2, torch::RestrictPtrTraits>(),
             rays_d.packed_accessor32<float, 2, torch::RestrictPtrTraits>(),
             hits_t.packed_accessor32<float, 2, torch::RestrictPtrTraits>(),
-            alive_indices.packed_accessor64<long, 1, torch::RestrictPtrTraits>(),
+            alive_indices.packed_accessor64<int64_t, 1, torch::RestrictPtrTraits>(),
             density_bitfield.data_ptr<uint8_t>(),
             cascades,
             grid_size,
diff --git a/models/csrc/volumerendering.cu b/models/csrc/volumerendering.cu
index c0188b577..862ce280c 100644
--- a/models/csrc/volumerendering.cu
+++ b/models/csrc/volumerendering.cu
@@ -193,7 +193,7 @@ __global__ void composite_test_fw_kernel(
     const torch::PackedTensorAccessor<scalar_t, 2, torch::RestrictPtrTraits, size_t> deltas,
     const torch::PackedTensorAccessor<scalar_t, 2, torch::RestrictPtrTraits, size_t> ts,
     const torch::PackedTensorAccessor<scalar_t, 2, torch::RestrictPtrTraits, size_t> hits_t,
-    torch::PackedTensorAccessor64<long, 1, torch::RestrictPtrTraits> alive_indices,
+    torch::PackedTensorAccessor64<int64_t, 1, torch::RestrictPtrTraits> alive_indices,
     const scalar_t T_threshold,
     const torch::PackedTensorAccessor32<int, 1, torch::RestrictPtrTraits> N_eff_samples,
     torch::PackedTensorAccessor<scalar_t, 1, torch::RestrictPtrTraits, size_t> opacity,
@@ -258,7 +258,7 @@ void composite_test_fw_cu(
             deltas.packed_accessor<scalar_t, 2, torch::RestrictPtrTraits, size_t>(),
             ts.packed_accessor<scalar_t, 2, torch::RestrictPtrTraits, size_t>(),
             hits_t.packed_accessor<scalar_t, 2, torch::RestrictPtrTraits, size_t>(),
-            alive_indices.packed_accessor64<long, 1, torch::RestrictPtrTraits>(),
+            alive_indices.packed_accessor64<int64_t, 1, torch::RestrictPtrTraits>(),
             T_threshold,
             N_eff_samples.packed_accessor32<int, 1, torch::RestrictPtrTraits>(),
             opacity.packed_accessor<scalar_t, 1, torch::RestrictPtrTraits, size_t>(),