Add missing SH degree 1 in cpu pipeline

pierotofy · Jun 5, 2024 · 1ff3d05 · 1ff3d05
1 parent 5c557d7
commit 1ff3d05
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rasterizer/gsplat-cpu/gsplat_cpu.cpp b/rasterizer/gsplat-cpu/gsplat_cpu.cpp
@@ -439,6 +439,9 @@ torch::Tensor compute_sh_forward_tensor_cpu(
         torch::Tensor x = xyz[0];
         torch::Tensor y = xyz[1];
         torch::Tensor z = xyz[2];
+        result.index_put_({"...", 1}, SH_C1 * -y);
+        result.index_put_({"...", 2}, SH_C1 * z);
+        result.index_put_({"...", 3}, SH_C1 * -x);
 
         if (numBases > 4){
             torch::Tensor xx = x * x;