shared_ptr to unique_ptr

pytorch · Nov 28, 2023 · bbde87f · bbde87f
1 parent ab47b28
commit bbde87f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.