如何在tensorflow的分布中运用Tensorflow XLA AOT支持

时间:2017-03-10 13:05:12

标签: tensorflow

Tensorflow 1.0引入了XLA支持,包括JIT编译和AOT编译。对于JIT编译,我找到了一个python测试脚本,可以对其进行单元测试。但是,我没有找到任何AOT编译的python测试。但是有一些bazel测试,可以在源代码树上运行。 Tensorflow的链接https://www.tensorflow.org/performance/xla/tfcompile提供了有关如何测试的信息。但是tfcompile并没有成为张量流的分发内容。我可能在这里错了。但我无法在安装它的TF的分发目录中的任何地方看到tfcompile。 任何人都可以帮我理解如何在现有的发行内容上测试AOT编译,或者我需要在代码中调整一些内容以允许AOT内容进入发行版? 提前谢谢。

1 个答案:

答案 0 :(得分:0)

我知道您是专门询问AOT,但我建议您先阅读此页:https://www.tensorflow.org/performance/xla/

然后阅读本文:https://www.tensorflow.org/performance/xla/jit

特别注意XLA不包含在我们的二进制分布中;此刻必须从源代码构建。请注意,在运行./configure时必须选择“启用XLA”才能启用XLA支持。

一旦你这样做,Yaroslav Bulatov的建议是正确的;你可以自己构建二进制文件,或者通过bazel运行测试。