chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove jaxlib.hlo_helpers.

Browse Source 
This module has been deprecated for many months and its replacement is jax.ffi.

PiperOrigin-RevId: 808567797
Change-Id: I5b4f4eee5e65ccd86e597c5761e68c36292b6209
This commit is contained in:
Peter Hawkins
2025-09-18 07:17:16 -07:00
committed by Copybara-Service
parent f9f3dd3c1d
commit 0e61a4ac5e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/third_party/warp/jax_experimental/custom_call.py
Vendored
+1 -1
View File
@@ -119,7 +119,7 @@ def _create_jax_warp_primitive():
from jax._src.interpreters import batching
from jax.interpreters import mlir
from jax.interpreters.mlir import ir
from jax.jaxlib.hlo_helpers import custom_call
from jax.google.hlo_helpers import custom_call
global _jax_warp_p
global _cc_callback