FEX-Emu项目中AVX128指令的256位连续掩码加载存储优化
2025-06-30 08:13:38作者:胡唯隽
在FEX-Emu这个x86模拟器的开发过程中,开发团队发现了一个可以优化的性能点,特别是在处理AVX128指令集中的256位连续掩码加载和存储操作时。
问题背景
当前实现中,SVE(可伸缩向量扩展)的连续掩码加载操作会将指令拆分为两部分,中间插入一个地址生成的加法操作。具体表现为,在处理类似vmaskmovps ymm0, ymm1, [rax]这样的256位掩码加载指令时,模拟器会生成两条加载指令,并在它们之间插入一个地址加法操作。
现有实现分析
从示例代码可以看到,当前的实现流程如下:
- 首先加载一个128位的向量(q2)
- 保存状态寄存器(nzcv)
- 设置掩码条件(p0.s)
- 执行第一条128位掩码加载(ld1w)
- 计算下一个128位的地址(add)
- 设置第二个掩码条件
- 执行第二条128位掩码加载
- 恢复状态寄存器
- 存储结果
其中第5步的地址加法操作实际上是可以优化的,因为SVE的ld1w和ld1d指令本身就支持一个短立即数偏移量,这个偏移量可以乘以VL(向量长度)来自动计算地址。
优化方案
通过利用SVE指令的立即数偏移特性,可以消除中间的加法指令。具体来说:
- 第一条加载指令保持从基地址x4开始
- 第二条加载指令可以直接使用立即数偏移,如
ld1w {z2.s}, p0/z, [x4, #1, mul vl]
这样不仅减少了指令数量(从9条减到8条),还消除了一个依赖关系,使得两条加载指令可以更高效地并行执行。
影响范围
这个优化适用于:
- 256位的vmaskmovps(单精度浮点掩码加载)
- 256位的vmaskmovpd(双精度浮点掩码加载)
- 对应的存储操作(vmaskmovps和vmaskmovpd的存储版本)
虽然每条指令只节省了一条指令,但由于这些操作在向量化代码中频繁使用,累积起来的性能提升会相当可观。
实现难度
从技术实现角度来看,这个优化相对简单直接,主要工作是修改指令发射逻辑,利用现有的SVE指令特性。不需要引入新的指令或复杂的地址计算逻辑。
性能预期
虽然每条指令序列只减少了一条指令,但考虑到:
- 减少了流水线的压力
- 降低了指令缓存的使用
- 减少了寄存器间的数据依赖
- 在循环中这些操作会被反复执行
因此整体性能提升可能比简单的指令计数减少更为显著。
总结
通过对SVE指令特性的深入理解和利用,FEX-Emu项目团队发现并实现了一个简单而有效的优化,提升了AVX128指令集中256位连续掩码加载和存储操作的模拟效率。这种优化展示了在指令模拟器中,充分理解目标架构(这里是ARM的SVE)特性对于生成高效代码的重要性。
登录后查看全文
热门项目推荐
相关项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0172
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook097
Step-3.7-FlashStep-3.7-Flash是一个拥有 1980 亿参数的稀疏混合专家(MoE)视觉语言模型,由 1960 亿参数的语言主干网络和 18 亿参数的视觉编码器组合而成,具备原生图像理解能力。Python00
BitCPM-CANN-8BBitCPM-CANN 是首个基于华为昇腾 NPU 原生构建的端到端 1.58 位(三值化)大语言模型训练系统。该系统将量化感知训练(QAT)集成到 Megatron-LM 框架中,并结合 MindSpeed 加速,覆盖了从自定义三值算子到基于昇腾 910B 的分布式并行训练的完整训练栈。Python00
MiniCPM5-1BMiniCPM5-1B,这是 MiniCPM5 系列的首款模型。它是一个专为端侧、本地部署和资源受限场景打造的 10 亿参数密集型 Transformer 模型,达到了 10 亿参数级开源模型的 SOTA 水平Jinja00
skillhubopenJiuwen 生态的 Skill 托管与分发开源方案,支持自建与可选 ClawHub 兼容。Python0239
热门内容推荐
最新内容推荐
项目优选
收起
deepin linux kernel
C
32
16
暂无描述
Dockerfile
750
4.87 K
Claude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed.
Get Started
Rust
1.58 K
172
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
841
1.84 K
Ascend Extension for PyTorch
Python
689
834
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
229
97
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
451
418
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.02 K
1.04 K
暂无简介
Dart
999
259
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
642
1.27 K