Skip to content

Commit 0e61a4a

Browse files
hawkinspcopybara-github
authored andcommitted
Remove jaxlib.hlo_helpers.
This module has been deprecated for many months and its replacement is jax.ffi. PiperOrigin-RevId: 808567797 Change-Id: I5b4f4eee5e65ccd86e597c5761e68c36292b6209
1 parent f9f3dd3 commit 0e61a4a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mjx/mujoco/mjx/third_party/warp/jax_experimental/custom_call.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def _create_jax_warp_primitive():
119119
from jax._src.interpreters import batching
120120
from jax.interpreters import mlir
121121
from jax.interpreters.mlir import ir
122-
from jax.jaxlib.hlo_helpers import custom_call
122+
from jax.google.hlo_helpers import custom_call
123123

124124
global _jax_warp_p
125125
global _cc_callback

0 commit comments

Comments
 (0)