openxla / xla

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

[xla:cpu] Pass buffer allocations for arguments and results when emitting kernel prototype #14404

Closed copybara-service[bot] closed 2 weeks ago

copybara-service[bot] commented 2 weeks ago

[xla:cpu] Pass buffer allocations for arguments and results when emitting kernel prototype

Buffer allocations are needed for deriving LLVM aliasing scopes (coming next!).