Fix kernel order

Author: StafaH

mujoco_warp/_src/bvh.py

@@ -723,6 +723,7 @@ def _update_flex_face_points(
   nflex: int,
   flex_dim: wp.array(dtype=int),
   flex_vertadr: wp.array(dtype=int),
+  flex_elemdataadr: wp.array(dtype=int),
   flex_elemnum: wp.array(dtype=int),
   flex_shelldataadr: wp.array(dtype=int),
   flex_elem: wp.array(dtype=int),
@@ -732,7 +733,6 @@ def _update_flex_face_points(
   flexvert_xpos_in: wp.array2d(dtype=wp.vec3),
   # In:
   flexvert_norm_in: wp.array2d(dtype=wp.vec3),
-  flex_elemdataadr: wp.array(dtype=int),
   flex_faceadr: wp.array(dtype=int),
   flex_workadr: wp.array(dtype=int),
   flex_worknum: wp.array(dtype=int),
@@ -1004,14 +1004,14 @@ def refit_flex_bvh(m: Model, d: Data, rc: RenderContext):
       m.nflex,
       m.flex_dim,
       m.flex_vertadr,
+      m.flex_elemdataadr,
       m.flex_elemnum,
       m.flex_shelldataadr,
       m.flex_elem,
       m.flex_shell,
       m.flex_radius,
       d.flexvert_xpos,
       flexvert_norm,
-      rc.flex_elemdataadr,
       rc.flex_faceadr,
       rc.flex_workadr,
       rc.flex_worknum,