@@ -108,51 +108,51 @@ def _sync_particle_points(
108108def _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 )
122122def _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 )
136136def _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 )
148148def _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
158158class NewtonSceneDataBackend (SceneDataBackend ):
0 commit comments