Skip to content

Commit bfebd49

Browse files
committed
Refactor renderCUDA function to use const int* compute_locally_1D_2D_map in backward.cu
1 parent 75cc5a8 commit bfebd49

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

cuda_rasterizer/backward.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,7 @@ renderCUDA(
408408
const float* __restrict__ colors,
409409
const float* __restrict__ final_Ts,
410410
const uint32_t* __restrict__ n_contrib,
411-
const bool* __restrict__ compute_locally,
411+
const int* __restrict__ compute_locally_1D_2D_map,
412412
const float* __restrict__ dL_dpixels,
413413
float3* __restrict__ dL_dmean2D,
414414
float4* __restrict__ dL_dconic2D,
@@ -667,7 +667,7 @@ void BACKWARD::render(
667667
const float* colors,
668668
const float* final_Ts,
669669
const uint32_t* n_contrib,
670-
const bool* compute_locally,
670+
const int* compute_locally_1D_2D_map,
671671
const float* dL_dpixels,
672672
float3* dL_dmean2D,
673673
float4* dL_dconic2D,
@@ -684,7 +684,7 @@ void BACKWARD::render(
684684
colors,
685685
final_Ts,
686686
n_contrib,
687-
compute_locally,
687+
compute_locally_1D_2D_map,
688688
dL_dpixels,
689689
dL_dmean2D,
690690
dL_dconic2D,

cuda_rasterizer/backward.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ namespace BACKWARD
3131
const float* colors,
3232
const float* final_Ts,
3333
const uint32_t* n_contrib,
34-
const bool* compute_locally,
34+
const int* compute_locally_1D_2D_map,
3535
const float* dL_dpixels,
3636
float3* dL_dmean2D,
3737
float4* dL_dconic2D,

0 commit comments

Comments
 (0)