fix https://github.com/ashawkey/stable-dreamfusion/issues/17
Browse files
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,
|
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", ([&] {
|