executable_args, device,
            GetPjRtExecuteOptions(device_type,
                                  std::move(non_donatable_input_indices)),
            future));
  } else {
    TF_ASSIGN_OR_RETURN(
        execute_outputs,
        executable->ExecutePortable(
            executable_args, device,
            GetPjRtExecuteOptions(device_type,

  Tensor* expected = CreateHostTensor<int32>(TensorShape({1, 3}), {3, 4, 5});
  test::ExpectTensorEqual<int32>(*expected, *host_tensor);
}

TEST(XlaLaunchUtilTest, GetPjRtExecuteOptions) {
  xla::ExecuteOptions options =
      GetPjRtExecuteOptions(DeviceType(DEVICE_GPU), {});
  EXPECT_FALSE(options.arguments_are_tupled);
  EXPECT_TRUE(options.untuple_result);
  EXPECT_FALSE(options.strict_shape_checking);

    std::vector<std::unique_ptr<xla::PjRtBuffer>>& executable_outputs,
    OpKernelContext* ctx);

// Returns the options used for executing a PjRtLoadedExecutable.
xla::ExecuteOptions GetPjRtExecuteOptions(
    const DeviceType& device_type,
    absl::flat_hash_set<int> non_donatable_input_indices);

// Returns the device ordinal from the parsed name of the device.
int GetDeviceOrdinal(const DeviceBase* device);