Use LoadedExecutableRef instead of std::unique_ptr<LoadedExecutable>

junwhanahn · Google-ML-Automation · commit 35b25c914a62 · 2025-05-06T16:37:48.000-07:00
PiperOrigin-RevId: 755573315
diff --git a/jaxlib/py_client.cc b/jaxlib/py_client.cc
@@ -446,7 +446,7 @@ PyClient::CompileIfrtProgram(
     }
   }
 
-  std::unique_ptr<ifrt::LoadedExecutable> ifrt_loaded_executable;
+  ifrt::LoadedExecutableRef ifrt_loaded_executable;
   std::optional<std::string> fingerprint;
   {
     nb::gil_scoped_release gil_release;
@@ -529,7 +529,7 @@ PyClient::DeserializeExecutable(nb_class_ptr<PyClient> client,
                                 ifrt::DeviceListRef executable_devices,
                                 std::optional<CompileOptions> options,
                                 std::vector<nb::capsule> host_callbacks) {
-  std::unique_ptr<ifrt::LoadedExecutable> ifrt_loaded_executable;
+  ifrt::LoadedExecutableRef ifrt_loaded_executable;
   std::optional<std::string> fingerprint;
   auto ifrt_deserialize_options = MakeIfrtDeserializeExecutableOptions(
       std::move(options), std::move(executable_devices),

Original file line number	Diff line number	Diff line change
`@@ -446,7 +446,7 @@ PyClient::CompileIfrtProgram(`
`446`	`446`	`}`
`447`	`447`	`}`
`448`	`448`
`449`		`- std::unique_ptr<ifrt::LoadedExecutable> ifrt_loaded_executable;`
	`449`	`+ ifrt::LoadedExecutableRef ifrt_loaded_executable;`
`450`	`450`	`std::optional<std::string> fingerprint;`
`451`	`451`	`{`
`452`	`452`	`nb::gil_scoped_release gil_release;`
`@@ -529,7 +529,7 @@ PyClient::DeserializeExecutable(nb_class_ptr<PyClient> client,`
`529`	`529`	`ifrt::DeviceListRef executable_devices,`
`530`	`530`	`std::optional<CompileOptions> options,`
`531`	`531`	`std::vector<nb::capsule> host_callbacks) {`
`532`		`- std::unique_ptr<ifrt::LoadedExecutable> ifrt_loaded_executable;`
	`532`	`+ ifrt::LoadedExecutableRef ifrt_loaded_executable;`
`533`	`533`	`std::optional<std::string> fingerprint;`
`534`	`534`	`auto ifrt_deserialize_options = MakeIfrtDeserializeExecutableOptions(`
`535`	`535`	`std::move(options), std::move(executable_devices),`