mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/rocm
History
Sergei Lebedev 51fc4f85ad Ported LuPivotsToPermutation to the typed XLA FFI 
The typed FFI

* allows passing custom call attributes directly to backend_config= instead
  of serializing them into a C++ struct.
* It also handles validation and deserialization of custom call operands.

PiperOrigin-RevId: 630067005
2024-05-02 08:12:05 -07:00
..
BUILD.bazel
Ported LuPivotsToPermutation to the typed XLA FFI
2024-05-02 08:12:05 -07:00