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

This CL introduces 'PluginProgram' in IFRT and exposes this in python via xla_client.compile_ifrt_program().

Browse Source 
The IFRT `PluginProgram` is simply a wrapper for arbitrary byte-strings: an IFRT backend that recognizes `PluginProgram` can interpret the byte-string in any way it sees fit.

PiperOrigin-RevId: 621258245
This commit is contained in:
jax authors 2024-04-02 12:19:57 -07:00
parent 60458bb36a
commit fb55d59143
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jaxlib/tools/build_wheel.py
View File

@ -100,6 +100,7 @@ def patch_copy_mlir_import(src_file, dst_dir):
_XLA_EXTENSION_STUBS = [
"__init__.pyi",
"ifrt_programs.pyi",
"ifrt_proxy.pyi",
"jax_jit.pyi",
"ops.pyi",