CNIPA.AI
返回搜索
档案

一种张量程序的函数化方法、设备及程序产品

发明专利有效
10权利要求 · 3 独立
§ Ⅰ

卷宗概要

申请人

上海人工智能创新中心

发明人

麻津铭; 李秀红; 王梓涵; 陈戒非; 张行程; 颜深根; 陈雨亭; 金旻玺; 蒋丽娟; 梁云; 杨超; 林达华; 裴芝林; 王辉; 刘亮

IPC 分类

G06F 8/41 (2018.01)G06F 9/50 (2006.01)

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