Update ifrt_computation_client.cc

pytorch · Sep 13, 2024 · d14796d · d14796d
1 parent 09b5555
commit d14796d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/csrc/runtime/ifrt_computation_client.cc b/torch_xla/csrc/runtime/ifrt_computation_client.cc
@@ -320,7 +320,7 @@ ComputationClient::DataPtr IfrtComputationClient::TransferShardsToDevice(
     shard_shapes.push_back(ifrt_shard->buffer->shape());
   }
   xla::ifrt::Shape ifrt_shape(shape.dimensions());
-  xla::ifrt::DeviceList devices_list({client_->addressable_devices().begin(),
+  tsl::RCReference<xla::ifrt::DeviceList> devices_list = xla::ifrt::BasicDeviceList::Create({client_->addressable_devices().begin(),
                                       client_->addressable_devices().end()});
   std::unique_ptr<xla::ifrt::Sharding> ifrt_sharding =
       xla::ifrt::ConcreteSharding::Create(devices_list, xla::ifrt::MemoryKind(),