ashawkey commited on
Commit
69fbfd2
·
1 Parent(s): a0c4308

fix https://github.com/ashawkey/stable-dreamfusion/issues/17

Browse files
Files changed (1) hide show
  1. raymarching/src/raymarching.cu +1 -1
raymarching/src/raymarching.cu CHANGED
@@ -905,7 +905,7 @@ __global__ void kernel_composite_rays(
905
  }
906
 
907
 
908
- void composite_rays(const uint32_t n_alive, const uint32_t n_step, const float T_thresh, at::Tensor rays_alive, at::Tensor rays_t, const at::Tensor sigmas, const at::Tensor rgbs, const at::Tensor deltas, at::Tensor weights, at::Tensor depth, at::Tensor image) {
909
  static constexpr uint32_t N_THREAD = 128;
910
  AT_DISPATCH_FLOATING_TYPES_AND_HALF(
911
  image.scalar_type(), "composite_rays", ([&] {
 
905
  }
906
 
907
 
908
+ void composite_rays(const uint32_t n_alive, const uint32_t n_step, const float T_thresh, at::Tensor rays_alive, at::Tensor rays_t, at::Tensor sigmas, at::Tensor rgbs, at::Tensor deltas, at::Tensor weights, at::Tensor depth, at::Tensor image) {
909
  static constexpr uint32_t N_THREAD = 128;
910
  AT_DISPATCH_FLOATING_TYPES_AND_HALF(
911
  image.scalar_type(), "composite_rays", ([&] {