我如何运行进行最小的 XLA C++ 计算？

Question

我正在使用 XLA C++ API，并且我设法运行进行了简单的添加，但我不知道我是否做对了。似乎有很多类我没用过。这是我的例子

auto builder = new XlaBuilder("XlaBuilder");
auto one = ConstantR0(builder, 1);
auto two = ConstantR0(builder, 2);
auto res = one + two;

ValueInferenceMode value_inf_mode;
auto value_inf = new ValueInference(builder_);

auto lit = value_inf
    ->AnalyzeConstant(res, value_inf_mode)
    ->GetValue()
    ->Clone();

// I'm using `untyped_data` because I can't express arbitrary array types.
// I guess I could use `data<int32>` in this simple case
auto data = lit.untyped_data();

std::cout << ((int32*) data)[0] << std::endl;  // prints 3

Answer 1

我怀疑我实际上并没有运行通过 XLA 进行计算。这是一种基于 XLA 源代码

中的 sample harness 的不同方法

XlaComputation computation = res.builder()->Build().ConsumeValueOrDie();
ExecutionProfile profile;
Literal lit = ClientLibrary::LocalClientOrDie()
    ->ExecuteAndTransfer(computation, {}, nullptr, &profile)
    .ConsumeValueOrDie();

data = lit.untyped_data()

我如何运行进行最小的 XLA C++ 计算？

How do I run a minimal XLA C++ computation?

c++

tensorflow-xla

我如何 运行 进行最小的 XLA C++ 计算？

How do I run a minimal XLA C++ computation?

c++

tensorflow-xla

我如何运行进行最小的 XLA C++ 计算？