From f174b00f23cb0a402a6bfe682188e6063b72928b Mon Sep 17 00:00:00 2001 From: Changhui Lin Date: Mon, 17 Mar 2025 17:17:28 -0700 Subject: [PATCH] Replace the uses of `PjRtClient::Compile()` with `PjRtClient::CompileAndLoad()`. This is to prepare for updating `PjRtClient::Compile()` to return an unloaded executable [1/N] PiperOrigin-RevId: 737805623 --- examples/jax_cpp/main.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/jax_cpp/main.cc b/examples/jax_cpp/main.cc index defbf1938..0a1d3a63a 100644 --- a/examples/jax_cpp/main.cc +++ b/examples/jax_cpp/main.cc @@ -81,7 +81,7 @@ int main(int argc, char** argv) { xla::XlaComputation xla_computation(test_module_proto); xla::CompileOptions compile_options; std::unique_ptr executable = - client->Compile(xla_computation, compile_options).value(); + client->CompileAndLoad(xla_computation, compile_options).value(); // Prepare inputs. xla::Literal literal_x =