Skip to content

Commit c8841dc

Browse files
committed
Fix bugs for depth interface
1 parent f844a97 commit c8841dc

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

cuda_rasterizer/forward.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,13 @@ namespace FORWARD
5555
int W, int H,
5656
const float2* points_xy_image,
5757
const float* features,
58+
const float* depths,
5859
const float4* conic_opacity,
5960
float* final_T,
6061
uint32_t* n_contrib,
6162
const float* bg_color,
62-
float* out_color);
63+
float* out_color,
64+
float* out_depth);
6365
}
6466

6567

cuda_rasterizer/rasterizer.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ namespace CudaRasterizer
4949
const float tan_fovx, float tan_fovy,
5050
const bool prefiltered,
5151
float* out_color,
52+
float* out_depth,
5253
int* radii = nullptr);
5354

5455
static void backward(

0 commit comments

Comments
 (0)