fix the offset of K in projection packed fwd 2dgs kernel

nerfstudio-project · Feb 25, 2025 · c37a659 · c37a659
1 parent bd64a47
commit c37a659
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gsplat/cuda/csrc/fully_fused_projection_packed_2dgs_fwd.cu b/gsplat/cuda/csrc/fully_fused_projection_packed_2dgs_fwd.cu
@@ -93,6 +93,7 @@ __global__ void fully_fused_projection_packed_fwd_2dgs_kernel(
         // camera space
         quats += col_idx * 4;
         scales += col_idx * 3;
+        Ks += row_idx * 9;
 
         mat3<T> RS_camera =
             R * quat_to_rotmat<T>(glm::make_vec4(quats)) *