mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace the uses of PjRtClient::Compile() with PjRtClient::CompileAndLoad().

Browse Source 
This is to prepare for updating `PjRtClient::Compile()` to return an unloaded executable [1/N]

PiperOrigin-RevId: 737805623
This commit is contained in:
Changhui Lin 2025-03-17 17:17:28 -07:00 committed by jax authors
parent 8c35191725
commit f174b00f23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/jax_cpp/main.cc
View File

@ -81,7 +81,7 @@ int main(int argc, char** argv) {
xla::XlaComputation xla_computation(test_module_proto);
xla::CompileOptions compile_options;
std::unique_ptr<xla::PjRtLoadedExecutable> executable =
client->Compile(xla_computation, compile_options).value();
client->CompileAndLoad(xla_computation, compile_options).value();
// Prepare inputs.
xla::Literal literal_x =