Wave propagation based on the spectral element method (SEM) is a representative HPC workload, but existing SEM implementations are not well matched to emerging ARM multicore CPUs with Scalable Matrix Extension (SME). We present an SME-enabled optimization of \textsc{SPECFEM3D} on the emerging LX2 processor that combines an SME-aware batched small-matrix kernel for SEM tensor-product operators, a memory-aware hybrid MPI+OpenMP execution scheme for limited-HBM systems, and a dispersion-based iso-accuracy study of the $(h,p)$ tradeoff. At fixed polynomial order, the optimized implementation improves full-application performance by 4--6$\times$ over the original code and delivers clear gains over optimized non-SME CPU baselines. Beyond these implementation-level gains, our results suggest that SME shifts the performance-favorable operating point toward higher polynomial orders along the dispersion-based iso-accuracy frontier, further reducing time-to-solution and working-set size. These results indicate that SME affects not only kernel efficiency, but also the practical discretization tradeoff for SEM on modern ARM multicore platforms.


翻译:基于谱元法(SEM)的波传播计算是高性能计算(HPC)领域的典型负载,但现有SEM实现方案难以适配配备可扩展矩阵扩展(SME)的新兴ARM多核CPU。本文针对新兴LX2处理器提出了一种启用SME的优化方案,具体包括:面向SEM张量积算子的SME感知批处理小矩阵核、面向有限高带宽存储(HBM)系统的内存感知混合MPI+OpenMP执行框架,以及基于色散等精度分析的(h,p)权衡研究。在固定多项式阶数下,优化实现使完整应用性能较原始代码提升4-6倍,并相较优化后的非SME CPU基线方案展现显著优势。除执行效率提升外,研究结果揭示:沿色散等精度前沿,SME将性能优势工作点向更高多项式阶数偏移,进一步缩短求解时间并降低工作集规模。这表明SME不仅影响核心计算效率,更改变了现代ARM多核平台上SEM离散方案的实用权衡策略。

0
下载
关闭预览

相关内容

《高功率微波的军事应用》最新131页报告
专知会员服务
72+阅读 · 2023年7月12日
专知会员服务
19+阅读 · 2021年6月29日
综述:军事应用中使用的一些重要算法
专知
13+阅读 · 2022年7月3日
美军电磁频谱战的发展及启示
科技导报
19+阅读 · 2019年3月25日
论文笔记:多任务相关粒子滤波跟踪器
统计学习与视觉计算组
10+阅读 · 2017年7月7日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Arxiv
0+阅读 · 4月29日
VIP会员
最新内容
反制无人机:乌克兰提供的五点启示
专知会员服务
7+阅读 · 9月23日
《各指挥层级均亟需红队能力》报告
专知会员服务
8+阅读 · 9月23日
《航电任务系统框架(FAMOS)》50页报告
专知会员服务
4+阅读 · 9月22日
《对抗行动中的人工智能与自主性》智库报告
专知会员服务
9+阅读 · 9月22日
《从数据到胜利:战争中的分析优势之争》
专知会员服务
11+阅读 · 9月22日
战争不仅需要机器人:人类仍不可或缺
专知会员服务
6+阅读 · 9月21日
《描绘美国防部创新基础设施的未来蓝图》100页
专知会员服务
10+阅读 · 9月21日
相关基金
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员