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

Fix docs referring to --print-ir-before/after-all, which has changed to --mlir-print....

Browse Source 
PiperOrigin-RevId: 448890307
This commit is contained in:
Christian Sigg 2022-05-16 00:39:10 -07:00 committed by jax authors
parent 6a6605263d
commit f73f03e039
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/_src/iree.py
View File

@ -145,7 +145,7 @@ class IreeClient:
del compile_options # Ignored.
iree_binary = iree.compiler.compile_str(
computation, target_backends=["dylib"], input_type="mhlo",
# extra_args=["--print-ir-after-all"],
# extra_args=["--mlir-print-ir-after-all"],
# extended_diagnostics=True,
)
# Load it into the runtime.