From bbde87f20d2065cdf28ea28bcfc1b05ceedb092c Mon Sep 17 00:00:00 2001
From: Will Cromar <wcromar@google.com>
Date: Tue, 28 Nov 2023 19:58:14 +0000
Subject: [PATCH] shared_ptr to unique_ptr

---
 torch_xla/csrc/runtime/pjrt_computation_client.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_xla/csrc/runtime/pjrt_computation_client.h b/torch_xla/csrc/runtime/pjrt_computation_client.h
index 0d8c7629d768..a9a234d0c022 100644
--- a/torch_xla/csrc/runtime/pjrt_computation_client.h
+++ b/torch_xla/csrc/runtime/pjrt_computation_client.h
@@ -104,7 +104,7 @@ class PjRtComputationClient : public ComputationClient {
   };
 
  private:
-  std::shared_ptr<xla::PjRtClient> client_;
+  std::unique_ptr<xla::PjRtClient> client_;
   std::unique_ptr<XlaCoordinator> coordinator_;
   // global_ordinals_ tracks a map from PjRtDeviceId to the device's
   // dense global ordinal.