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", ([&] {
|