草庐IT

python - tensorflow 的 XLA_GPU 和 XLA_CPU 是什么

我可以列出gpu设备唱以下tensorflow代码:importtensorflowastffromtensorflow.python.clientimportdevice_libprint(device_lib.list_local_devices())结果是:[name:"/device:CPU:0"device_type:"CPU"memory_limit:268435456locality{}incarnation:17897160860519880862,name:"/device:XLA_GPU:0"device_type:"XLA_GPU"memory_limit:171

程序分析与优化 - 9 附录 XLA的缓冲区指派

本章是系列文章的案例学习,不属于正篇,主要介绍了TensorFlow引入的XLA的优化算法。XLA也有很多局限性,XLA更多的是进行合并,但有时候如果参数特别多的场景下,也需要进行分割。XLA没有数据切分的功能。当前最主流的编译器领域的编译优化功能还是mlir。本文中的所有内容来自学习DCC888的学习笔记或者自己理解的整理,如需转载请注明出处。周荣华@燧原科技9.1 什么是XLAXLA是AcceleratedLinearAlgebra的简称。第一次看到Accelerated被简称为X的时候,有点奇怪,因为Accelerated里面可没有一个字母是X,但Accelerated的发音和X相同,这

程序分析与优化 - 9 附录 XLA的缓冲区指派

本章是系列文章的案例学习,不属于正篇,主要介绍了TensorFlow引入的XLA的优化算法。XLA也有很多局限性,XLA更多的是进行合并,但有时候如果参数特别多的场景下,也需要进行分割。XLA没有数据切分的功能。当前最主流的编译器领域的编译优化功能还是mlir。本文中的所有内容来自学习DCC888的学习笔记或者自己理解的整理,如需转载请注明出处。周荣华@燧原科技9.1 什么是XLAXLA是AcceleratedLinearAlgebra的简称。第一次看到Accelerated被简称为X的时候,有点奇怪,因为Accelerated里面可没有一个字母是X,但Accelerated的发音和X相同,这