[xla] Migrate to PjRtFuture<>::MakePromise() API

PiperOrigin-RevId: 806042342

Author: ezhulenev

third_party/xla/xla/pjrt/cpu/raw_buffer.cc

@@ -77,20 +77,19 @@ void CpuTrackedDeviceEventPromise::SetReady() {
 }
 
 PjRtFuture<> CpuTrackedDeviceEvent::GetReadyFuture() {
-  PjRtFuture<>::Promise promise = PjRtFuture<>::CreatePromise();
-  event_.AndThen([promise, event = event_]() mutable {
+  auto [promise, future] = PjRtFuture<>::MakePromise();
+  event_.AndThen([promise = std::move(promise), event = event_]() mutable {
     if (auto* error = event.GetErrorIfPresent()) {
       promise.Set(*error);
     } else {
       promise.Set();
     }
   });
 
-  return PjRtFuture<>(
-      promise,
+  return PjRtFutureHelpers::WithProfiling(
+      std::move(future),
       /*on_block_start=*/
-      [ready_event = FormRef(promise.async_value()),
-       callee_method = callee_method_, callee_type = callee_type_]() {
+      [callee_method = callee_method_, callee_type = callee_type_]() {
         tsl::profiler::TraceMeProducer traceme(
             [&] { return absl::StrCat(callee_type, "::", callee_method); });
         return PjRtFutureHelpers::ProfilingKeys({traceme.GetContextId()});