Update render.py

Author: tzole1155

moai/monads/render/nvdiffrast/render.py

@@ -25,7 +25,7 @@ def __init__(
         self.resolution = [height, width]
         self.decomposed = decomposed
 
-    # @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
+    @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
     def forward(
         self,
         ndc_vertices: torch.Tensor,
@@ -98,7 +98,7 @@ def __init__(
     ):
         super().__init__()
 
-    # @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
+    @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
     def forward(
         self,
         attributes: torch.Tensor,
@@ -125,7 +125,7 @@ def __init__(
         super().__init__()
         self.pos_grad_boost = position_gradient_scale
 
-    # @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
+    @torch.amp.custom_fwd(device_type="cuda", cast_inputs=torch.float32)
     def forward(
         self,
         attributes: torch.Tensor,