search

openxla / xla

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

[xla:ffi] Add ExecutionStage to FFI handler and check it at run time #14864

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

[xla:ffi] Add ExecutionStage to FFI handler and check it at run time

In preparation for implementing "stateful custom calls" add execution stage to FFI handler.