Merge pull request NVlabs#28 from zjjMaiMai/master

[fix] vren build error on windows cause long
nfyfamr · Jul 24, 2022 · 2c4d5da · 2c4d5da
2 parents bede6be + 2d20cc4
commit 2c4d5da
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git 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
@@ -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
@@ -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>(),