From 1ff3d0592a6fdfa6f980201fea65728a3a11c67e Mon Sep 17 00:00:00 2001 From: Willyzw Date: Wed, 5 Jun 2024 13:16:32 +0200 Subject: [PATCH] Add missing SH degree 1 in cpu pipeline --- rasterizer/gsplat-cpu/gsplat_cpu.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/rasterizer/gsplat-cpu/gsplat_cpu.cpp b/rasterizer/gsplat-cpu/gsplat_cpu.cpp index 9385265..694269b 100644 --- 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;