一种张量程序的函数化方法、设备及程序产品
本发明公开了一种张量程序的函数化方法、设备及程序产品。该方法包括:获取张量程序,并构建张量程序的图级中间表示程序;基于图级中间表示程序的内存依赖图和不可变算子对图级中间表示程序进行突变重写和张量版本关系标注得到重写图级中间表示程序;不可变算子用于替代图级中间表示程序中的突变算子,且不可变算子输出的新版本张量与输入的源张量拥有独立内存;对重写图级中间表示程序进行程序优化,得到函数式的基于静态单赋值的张量中间表示程序。将Graph‑Level IR程序与静态单赋值SSA结合,彻底消除了Graph‑Level IR程序中张量突变副作用,提升了深度学习编译器对张量程序的优化性能。