如何在张量流的分布中行使 Tensorflow XLA AOT 支持



TensorFlow 1.0引入了XLA支持,其中包括JIT编译和AOT汇编。对于JIT编译,我找到了一个可以通过单位测试的Python测试脚本。但是,我没有发现任何用于AOT汇编的Python测试。但是有Bazel测试,可以在源树上运行。Tensorflow的链接https://www.tensorflow.org/performance/xla/tfcompile提供有关如何测试的信息。但是TFCompile不会介入TensorFlow的分布内容中。我在这里可能错了。但是我看不到TF Comply在安装它的位置的TFCompile。谁能帮助我了解如何在现有的分发内容上测试AOT编译,或者我需要在代码中调整某些内容以允许AOT内容进行分发?预先感谢。

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

然后阅读此内容:https://www.tensorflow.org/performance/xla/jit

特别指的是,XLA不包含在我们的二进制分布中;您必须目前从源构建。请注意,运行./configure时必须选择"启用XLA"才能启用XLA支持。

完成此操作后,Yaroslav Bulatov的建议是正确的;您可以自己构建二进制文件,或通过Bazel进行测试。

最新更新