10.3969/j.issn.1007-130X.2023.05.003
基于监督学习的稀疏矩阵自动任务分配
针对稀疏矩阵与稠密向量乘运算探讨了不同的任务分配策略对性能的影响,观察到任务分配策略的选择会显著地影响稀疏矩阵的运算性能,且不存在一种固定的任务分配策略针对所有的稀疏矩阵都能获得最佳性能.为此,提出了一种基于机器学习的最优任务分配策略选择模型,其训练过程仅使用稀疏矩阵的特征来刻画输入数据集,且能够针对给定的数据集和目标平台自动地训练模型.实验结果表明,相对于默认的块分配方法,使用该模型选择的任务分配方式能够获得平均约35%的性能提升.
稀疏矩阵向量乘、任务分配、机器学习
45
TP391(计算技术、计算机技术)
国家自然科学基金;国家自然科学基金
2023-06-06(万方平台首次上网日期,不代表论文的发表时间)
共8页
782-789