diff --git a/gsplat/cuda/csrc/backward.cuh b/gsplat/cuda/csrc/backward.cuh index 14f983664..d4bc04c5c 100644 --- a/gsplat/cuda/csrc/backward.cuh +++ b/gsplat/cuda/csrc/backward.cuh @@ -1,5 +1,6 @@ #include #include +#include // for f : R(n) -> R(m), J in R(m, n), // v is cotangent in R(m), e.g. dL/df in R(m), diff --git a/gsplat/cuda/csrc/forward.cuh b/gsplat/cuda/csrc/forward.cuh index 168135c04..8cc08279d 100644 --- a/gsplat/cuda/csrc/forward.cuh +++ b/gsplat/cuda/csrc/forward.cuh @@ -1,5 +1,6 @@ #include #include +#include // compute the 2d gaussian parameters from 3d gaussian parameters void project_gaussians_forward_impl( diff --git a/gsplat/cuda/csrc/helpers.cuh b/gsplat/cuda/csrc/helpers.cuh index f3813ad0d..dd5be7e7b 100644 --- a/gsplat/cuda/csrc/helpers.cuh +++ b/gsplat/cuda/csrc/helpers.cuh @@ -1,5 +1,5 @@ #include "config.h" -#include "cuda_runtime.h" +#include #include "third_party/glm/glm/glm.hpp" #include "third_party/glm/glm/gtc/type_ptr.hpp" #include diff --git a/gsplat/cuda/csrc/serial_backward.cuh b/gsplat/cuda/csrc/serial_backward.cuh index 58b5930f3..bbb516049 100644 --- a/gsplat/cuda/csrc/serial_backward.cuh +++ b/gsplat/cuda/csrc/serial_backward.cuh @@ -1,4 +1,5 @@ -#include "cuda_runtime.h" +#include +#include __host__ __device__ float3 projectMean2DBackward( const float3 m, const float *proj, const float2 dL_dmean2D diff --git a/gsplat/cuda/csrc/sh.cuh b/gsplat/cuda/csrc/sh.cuh index 06946de4d..d51c18ac6 100644 --- a/gsplat/cuda/csrc/sh.cuh +++ b/gsplat/cuda/csrc/sh.cuh @@ -1,4 +1,4 @@ -#include "cuda_runtime.h" +#include #include namespace cg = cooperative_groups;