Skip to content

Commit c44042a

Browse files
committed
Raise exception on BatchRenderer when not using CUDA as backend.
1 parent 9e1ac5f commit c44042a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

genesis/vis/batch_renderer.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,12 +71,16 @@ def build(self):
7171
"""
7272
if len(self._visualizer._cameras) == 0:
7373
raise ValueError("No cameras to render")
74+
75+
if gs.backend != gs.cuda:
76+
gs.raise_exception("BatchRenderer requires CUDA backend.")
77+
7478
cameras = self._visualizer._cameras
7579
lights = self._lights
7680
rigid = self._visualizer.scene.rigid_solver
77-
device = torch.cuda.current_device()
7881
n_envs = self._visualizer.scene.n_envs if self._visualizer.scene.n_envs > 0 else 1
7982
res = cameras[0].res
83+
gpu_id = gs.device.index
8084
use_rasterizer = self._renderer_options.use_rasterizer
8185

8286
# Cameras
@@ -96,7 +100,7 @@ def build(self):
96100
light_cutoff = self.light_cutoff_tensor
97101

98102
self._renderer = MadronaBatchRendererAdapter(
99-
rigid, device, n_envs, n_cameras, n_lights, camera_fov, *res, False, use_rasterizer
103+
rigid, gpu_id, n_envs, n_cameras, n_lights, camera_fov, *res, False, use_rasterizer
100104
)
101105
self._renderer.init(
102106
rigid,

0 commit comments

Comments
 (0)