xla
xla copied to clipboard
openxla
A machine learning compiler for GPUs, CPUs, and ML accelerators
Sets the flag --experimental_link_static_libraries=true.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "copybara-service[bot] avatar"){kind=avatar}
Remove pointer to previous Python frame on Traceback frame. This removes Traceback->Frame->f_back. The context for this change: I want to attach Tracebacks to JAX run-time errors (checkify) that point to...
Make the default implementation of Log/Log2/Ln/Log1p in XLA accurate on all platforms, and tighten test tolerances accordingly. The new implementation is accurate to 2 ULP. This change also adds tests...
Updates Bazel version from 5.3.0 to 6.0.0
Currently bazel builds of XLA from the OpenXLA repository require disabling visibility checks via the `--nocheck_visibility` argument to Bazel. It would be great if we could fix this!
[CHLO] Implement EighOp and lowering to MHLO
[XLA] Take advantage of moveable host callbacks and reduce the number of allocations.
[PJRT:C] Add PJRT_Client_Defragment Also adds some helper functions: * PJRT_RETURN_STATUS_IF_ERROR (macro) * AwaitEvent * DefragmentClient * GetHostBuffer [JAX] Modify JaxArrayTest.test_defragment to work on any numbers of devices