message: [PJRT] Don't mention replica/partition numbers when reporting errors about computations with only a single local replica.
  Should improve error messages in cases like
diff: @@ -2029,6 +2029,9 @@ PjRtStreamExecutorExecutable::Execute(
  const int partition = addressable_device_logical_ids_[i].partition;
  auto& statusor = results[i];
  if (!statusor.ok()) {
  + if (num_addressable_devices == 1) {
  + return statusor.status();
  + } else {
  return AppendStatus(
  statusor.status(),
  absl::StrFormat("while running replica %d and partition %d of a "
  @@ -2036,6 +2039,7 @@ PjRtStreamExecutorExecutable::Execute(
  "replicas may have failed as well).",
  replica, partition));
  }
  + }
  wrapped_results[i] = std::move(statusor.ValueOrDie());
  }
  return wrapped_results;
  