Skip to content

Commit 4e636bd

Browse files
committed
Refactor world_mask data type from int32 to bool
1 parent 23a00ef commit 4e636bd

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

source/isaaclab_newton/isaaclab_newton/physics/newton_manager.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -108,51 +108,51 @@ def _sync_particle_points(
108108
def _or_reset_masks_from_mask(
109109
env_mask: wp.array(dtype=wp.bool),
110110
articulation_ids: wp.array2d(dtype=int),
111-
world_mask: wp.array(dtype=wp.int32),
111+
world_mask: wp.array(dtype=wp.bool),
112112
fk_mask: wp.array(dtype=wp.bool),
113113
):
114114
"""OR env_mask into world_mask and set corresponding articulation bits in fk_mask."""
115115
world, arti = wp.tid()
116116
if env_mask[world]:
117-
world_mask[world] = wp.int32(1)
117+
world_mask[world] = True
118118
fk_mask[articulation_ids[world, arti]] = True
119119

120120

121121
@wp.kernel(enable_backward=False)
122122
def _scatter_reset_masks_from_ids(
123123
env_ids: wp.array(dtype=int),
124124
articulation_ids: wp.array2d(dtype=int),
125-
world_mask: wp.array(dtype=wp.int32),
125+
world_mask: wp.array(dtype=wp.bool),
126126
fk_mask: wp.array(dtype=wp.bool),
127127
):
128128
"""Scatter-set world_mask and fk_mask from sparse env_ids."""
129129
i, arti = wp.tid()
130130
world = env_ids[i]
131-
world_mask[world] = wp.int32(1)
131+
world_mask[world] = True
132132
fk_mask[articulation_ids[world, arti]] = True
133133

134134

135135
@wp.kernel(enable_backward=False)
136136
def _or_world_mask_from_env_mask(
137137
env_mask: wp.array(dtype=wp.bool),
138-
world_mask: wp.array(dtype=wp.int32),
138+
world_mask: wp.array(dtype=wp.bool),
139139
):
140140
"""OR env_mask into world_mask. Used when articulation_ids is unavailable
141141
(e.g. closed-loop assets whose root view does not expose them)."""
142142
world = wp.tid()
143143
if env_mask[world]:
144-
world_mask[world] = wp.int32(1)
144+
world_mask[world] = True
145145

146146

147147
@wp.kernel(enable_backward=False)
148148
def _scatter_world_mask_from_ids(
149149
env_ids: wp.array(dtype=int),
150-
world_mask: wp.array(dtype=wp.int32),
150+
world_mask: wp.array(dtype=wp.bool),
151151
):
152152
"""Scatter-set world_mask from sparse env_ids. Used when articulation_ids
153153
is unavailable (e.g. closed-loop assets whose root view does not expose them)."""
154154
i = wp.tid()
155-
world_mask[env_ids[i]] = wp.int32(1)
155+
world_mask[env_ids[i]] = True
156156

157157

158158
class NewtonSceneDataBackend(SceneDataBackend):

0 commit comments

Comments
 (0)