mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 00:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
12,468 Commits 508 Branches 263 Tags
Commit Graph

11 Commits

Author SHA1 Message Date
Mehdi Amini
5a6cb438e8 Move MHLO to XLA 
As part of the OpenXLA project, we're splitting XLA outside of TensorFlow.
MHLO belongs to OpenXLA and we're relocating it nested under XLA to allow the
split. Some further directory layout change will likely happen over time.

PiperOrigin-RevId: 464126676
 2022-07-29 11:54:51 -07:00
Parker Schuh
d8f0099f68 _mlirTransforms merged into _mlirRegisterEverything. 
PiperOrigin-RevId: 462233907
 2022-07-20 14:43:27 -07:00
Peter Hawkins
e63765a7a6 Use symlink_files() to add version.py to jaxlib, rather than copying it in as part of the wheel assembly process. 
Change in preparation for supporting running JAX tests under Bazel. This change allows the Bazel py_library() to see version.py.

Update symlink_files Bazel macro to a newer version.

PiperOrigin-RevId: 458481396
 2022-07-01 09:07:03 -07:00
Robert Suderman
499a4e733c Expose ml_program dialect for MLIR builder 
We now have an ml_program dialect that describes global variables
including load and store operations. Expose this dialect to allow
exporting variables and constants.
 2022-06-28 20:29:41 +00:00
Yash Katariya
95a854d487 Rename the file to .cc to fix the broken builds 
PiperOrigin-RevId: 453941301
 2022-06-09 09:18:07 -07:00
Peter Hawkins
d56601a896 Include mlir.transforms and mlir.passmanager in jaxlib BUILD. 
These increase the binary size of jaxlib only a negligible amount, and allow running passes like canonicalization from Python.
 2022-05-16 13:00:22 +00:00
Aart Bik
c1261ccd27 Adds a wrapper to sparse tensor dialect, as part of an 
an initial prototype of an alternate JAX compilation path
that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO
together with sparse tensor types.

PiperOrigin-RevId: 443438043
 2022-04-21 11:48:44 -07:00
jax authors
6c45969fe4 Integrate LLVM at llvm/llvm-project@eb27da7dec 
Updates LLVM usage to match
[eb27da7dec67](https://github.com/llvm/llvm-project/commit/eb27da7dec67)

PiperOrigin-RevId: 432199388
 2022-03-03 08:24:39 -08:00
Cloud Han
317edcdacd fix mlir capi dll building and linking 2021-11-25 00:07:25 +08:00
Peter Hawkins
ce7ae6bd76 Make MLIR bindings build work under Bazel. 
Tested on Linux and Mac, but not Windows.
 2021-11-12 12:16:32 -05:00
Peter Hawkins
11f6c535ae Add MLIR:Python bindings to jaxlib build. 
PiperOrigin-RevId: 407657331
 2021-11-04 13:29:58 -07:00