Skip to content

Commit cc7685b

Browse files
authored
Disables GPU testing for suction gripper environments (#3373)
# Description Since suction gripper requires CPU simulation currently, we disable GPU environment testing for them for now and explicitly sets the device for these environments to CPU. ## Type of change <!-- As you go through the list, delete the ones that are not applicable. --> - Bug fix (non-breaking change which fixes an issue) ## Checklist - [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with `./isaaclab.sh --format` - [x] I have made corresponding changes to the documentation - [x] My changes generate no new warnings - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] I have updated the changelog and the corresponding version in the extension's `config/extension.toml` file - [ ] I have added my name to the `CONTRIBUTORS.md` or my name already exists there <!-- As you go through the checklist above, you can mark something as done by putting an x character in it For example, - [x] I have done this task - [ ] I have not done this task -->
1 parent 1a71e24 commit cc7685b

File tree

3 files changed

+11
-1
lines changed

3 files changed

+11
-1
lines changed

source/isaaclab/test/assets/test_surface_gripper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def generate_surface_gripper(
114114
for i in range(num_surface_grippers):
115115
prim_utils.create_prim(f"/World/Env_{i}", "Xform", translation=translations[i][:3])
116116
articulation = Articulation(articulation_cfg.replace(prim_path="/World/Env_.*/Robot"))
117-
surface_gripper_cfg = surface_gripper_cfg.replace(prim_expr="/World/Env_.*/Robot/Gripper/SurfaceGripper")
117+
surface_gripper_cfg = surface_gripper_cfg.replace(prim_path="/World/Env_.*/Robot/Gripper/SurfaceGripper")
118118
surface_gripper = SurfaceGripper(surface_gripper_cfg)
119119

120120
return surface_gripper, articulation, translations

source/isaaclab_tasks/isaaclab_tasks/manager_based/manipulation/stack/config/ur10_gripper/stack_joint_pos_env_cfg.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,9 @@ def __post_init__(self):
140140
# post init of parent
141141
super().__post_init__()
142142

143+
# Suction grippers currently require CPU simulation
144+
self.device = "cpu"
145+
143146
# Set events
144147
self.events = EventCfgLongSuction()
145148

@@ -178,6 +181,9 @@ def __post_init__(self):
178181
# post init of parent
179182
super().__post_init__()
180183

184+
# Suction grippers currently require CPU simulation
185+
self.device = "cpu"
186+
181187
# Set UR10 as robot
182188
self.scene.robot = UR10_SHORT_SUCTION_CFG.replace(prim_path="{ENV_REGEX_NS}/Robot")
183189

source/isaaclab_tasks/test/env_test_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,10 @@ def _run_environments(
114114
if isaac_sim_version < 5 and create_stage_in_memory:
115115
pytest.skip("Stage in memory is not supported in this version of Isaac Sim")
116116

117+
# skip suction gripper environments as they require CPU simulation and cannot be run with GPU simulation
118+
if "Suction" in task_name and device != "cpu":
119+
return
120+
117121
# skip these environments as they cannot be run with 32 environments within reasonable VRAM
118122
if num_envs == 32 and task_name in [
119123
"Isaac-Stack-Cube-Franka-IK-Rel-Blueprint-v0",

0 commit comments

Comments
 (0)