tensorflow MLIR backend for XLA

iovurdzv  于 5个月前  发布在  其他
关注(0)|答案(7)|浏览(55)

你好,我可以问几个关于XLA的简单问题。

  1. 是否有一个开源的XLA MLIR后端,可以尝试"开箱即用"?
  2. 是否有方法可以启用/禁用XLA中使用的各种优化?
  3. 了解XLA中可用优化的最佳方法是什么?
omqzjyyz

omqzjyyz1#

请查看XLA link,并告知我们是否有所帮助。谢谢!

bis0qfac

bis0qfac2#

@sushreebarsa ,这个链接非常简略,没有回答上述任何问题。因此,我打开了这个问题。

hsgswve4

hsgswve43#

你可以参考Tensorflow代码XLA和实现代码MLIR-HLO
关于XLA MLIR优化细节的详细信息,你可以参考this文档。如果这对你有帮助,请告诉我们。谢谢!

gupuwyp2

gupuwyp24#

谢谢。链接MLIR CodeGen for XLA看起来像是一个尚未实现的路线图。如果可能的话,能否澄清当前的高级别XLA流程和未来的MLIR流程?据我所知,当前的流程如下:Grappler(pbb ?)-> XLA HLO(独立)-XLA HLO(目标依赖)->LLVM IR->PTX->thunk
沿着相同的线路,基于MLIR的流程会是什么样子?
此外,似乎没有任何文档提到任何标志来启用/禁用优化类别。是否有可能启用/禁用某些优化类别?

093gszye

093gszye5#

@sachinprasadhs ,关于这个有什么更新吗?

ca1c2owp

ca1c2owp7#

链接MLIR CodeGen for XLA看起来像是一个尚未实施的路线图。
了解我们在实施这个路线图方面目前的位置也很好。

相关问题