mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14554 from hawkinsp:mlirapi

Browse Source 
PiperOrigin-RevId: 510410342
This commit is contained in:
jax authors 2023-02-17 05:53:33 -08:00
commit c62462afda
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/interpreters/mlir.py
View File

@ -62,5 +62,6 @@ from jax._src.interpreters.mlir import (
register_constant_handler as register_constant_handler,
register_lowering as register_lowering,
shape_tensor as shape_tensor,
token_type as token_type,
xla_computation_to_mlir_module as xla_computation_to_mlir_module,
)