Merge pull request #139 from zctu/main

Fix nan problem with Metal
pierotofy · Nov 17, 2024 · 57201dc · 57201dc
2 parents eba9476 + 53103c4
commit 57201dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rasterizer/gsplat-metal/gsplat_metal.metal b/rasterizer/gsplat-metal/gsplat_metal.metal
@@ -959,7 +959,7 @@ kernel void rasterize_backward_kernel(
             float2 v_xy_local = {0.f, 0.f};
             float v_opacity_local = 0.f;
             //initialize everything to 0, only set if the lane is valid
-            if(valid){
+            if(valid && alpha<0.99f){
                 // compute the current T for this gaussian
                 float ra = 1.f / (1.f - alpha);
                 T *= ra;
@@ -1430,4 +1430,4 @@ kernel void compute_cov2d_bounds_kernel(
     conics[index + 1] = conic.y;
     conics[index + 2] = conic.z;
     radii[row] = radius;
-}
+}