remove unnecessary .cuda()

nerfstudio-project · Dec 8, 2023 · c9e4b98 · c9e4b98
1 parent bade644
commit c9e4b98
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/gsplat/rasterize.py b/gsplat/rasterize.py
@@ -144,15 +144,15 @@ def backward(ctx, v_out_img, v_out_alpha=None):
             img_width,
             gaussian_ids_sorted.contiguous(),
             tile_bins,
-            xys.contiguous().cuda(),
-            conics.contiguous().cuda(),
-            colors.contiguous().cuda(),
-            opacity.contiguous().cuda(),
-            background.contiguous().cuda(),
-            final_Ts.contiguous().cuda(),
-            final_idx.contiguous().cuda(),
-            v_out_img.contiguous().cuda(),
-            v_out_alpha.contiguous().cuda(),
+            xys.contiguous(),
+            conics.contiguous(),
+            colors.contiguous(),
+            opacity.contiguous(),
+            background.contiguous(),
+            final_Ts.contiguous(),
+            final_idx.contiguous(),
+            v_out_img.contiguous(),
+            v_out_alpha.contiguous(),
         )
 
         return (