目录
4.1 XLA(Accelerated Linear Algebra)编译器架构
软件栈与编译优化
4.1 XLA(Accelerated Linear Algebra)编译器架构
XLA编译器采用分层中间表示(IR)架构,前端接收TensorFlow或JAX计算图后,将其转换为高级运算中间表示(HLO)。HLO IR采用静态单赋值(SSA)形式,每个张量值在程序文本中仅被定义一次,定义后即可用性(def-use)关系通过数据流边显式表达。IR层级由HloModule、HloComputation与HloInstruction三级嵌套结构组成:HloModule代表完整程序,包含全局配置与入口计算;HloComputation对应函数语义,支持递归调用与嵌套;HloInstruc
订阅专栏 解锁全文
505



被折叠的 条评论
为什么被折叠?



