面向TensorFlow和PyTorch的线性代数基准测试

作     者：胥凌 XU Ling

作者机构：航空工业西安航空计算技术研究所陕西西安710000 

基　　金：航空科学基金项目资助(2018ZC31003) 

出 版 物：《航空计算技术》 (Aeronautical Computing Technique)

年 卷 期：2022年第52卷第3期

页      码：5-9页

摘      要：传统的高性能线性代数计算库如BLAS需要开发者具备丰富的性能优化经验,使用困难。TensorFlow、Pytorch等AI框架提供了简单的开发接口,促进了机器学习应用的发展。这些AI框架大量进行线性代数计算,但是不清楚其是否针对线性代数计算进行了性能优化。设计了一组线性代数计算测试程序,评估了AI框架对的线性代数计算的优化程度。分析显示AI框架在计算图模型下可以有效去除冗余子表达式,但仍然缺少自动识别矩阵链最佳括号的相关优化。未来AI框架可以通过吸收现有高性能线性代数加速库的优化技术进一步提升性能。

主 题 词：线性代数 TensorFlow Pytorch 高性能计算 矩阵乘法 

学科分类：12[管理学] 1201[管理学-管理科学与工程类] 081104[081104] 08[工学] 0835[0835] 0811[工学-水利类] 0812[工学-测绘类] 

D　O　I：10.3969/j.issn.1671-654X.2022.03.002

馆 藏 号：203111764...