search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.39k stars 356 forks source link

[xla:cpu] Add FFI custom call thunk runtime support to PJRT CPU client. #14288

Closed copybara-service[bot] closed 2 days ago

copybara-service[bot] commented 2 days ago

[xla:cpu] Add FFI custom call thunk runtime support to PJRT CPU client.

Also add a benchmark that uses PJRT CPU Client.