num_sh_bases() is used in both host and device (#41)

Co-authored-by: Ruilong Li <[email protected]>
nerfstudio-project · Oct 13, 2023 · 2d65152 · 2d65152
1 parent b6f0a9b
commit 2d65152
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gsplat/cuda/csrc/sh.cuh b/gsplat/cuda/csrc/sh.cuh
@@ -30,7 +30,8 @@ __device__ const float SH_C4[] = {
     -1.7701307697799304f,
     0.6258357354491761f};
 
-__device__ unsigned num_sh_bases(const unsigned degree) {
+// This function is used in both host and device code
+__host__ __device__ unsigned num_sh_bases(const unsigned degree) {
     if (degree == 0)
         return 1;
     if (degree == 1)