FPMR寄存器详解

ARM浮点模式控制寄存器 - 深入解析与应用场景

FPMR是什么？

FPMR（Floating-point Mode Register，浮点模式寄存器）是ARM架构中的一个系统寄存器，主要用于控制FP8（8位浮点数）指令的行为。简单来说，它就是FP8操作的"控制中心"，决定了FP8数据如何被处理、转换和运算。

FPMR的主要功能

格式控制

指定FP8输入和输出的数据格式（E5M2或E4M3）

缩放控制

控制浮点数转换时的指数缩放值

溢出处理

定义当发生溢出时是生成无穷大还是最大正常数

什么情况下需要使用FPMR？

当你需要使用FP8格式进行浮点运算时，就需要配置FPMR寄存器。FP8格式相比传统的FP32/FP16格式，能够大幅减少内存占用和带宽需求，特别适合以下场景：

FPMR寄存器字段详解

F8S1, F8S2, F8D - 格式选择字段

这些字段用于选择FP8数据的格式：

字段名 位范围 功能 可选值
F8S1 [2:0] 第一个FP8输入数据流的格式 000: E5M2格式, 001: E4M3格式
F8S2 [5:3] 第二个FP8输入数据流的格式 000: E5M2格式, 001: E4M3格式
F8D [8:6] 输出结果的FP8格式 000: E5M2格式, 001: E4M3格式

注意：设置不支持的格式值会导致不可预测的行为，软件需要先检查ID_AA64FPFR0_EL1[7:0]确认格式支持情况。

OSM, OSC - 溢出饱和控制

这些位控制溢出时的处理方式：

字段名 位位置 功能 取值含义
OSM [14] FP8乘法指令溢出饱和 0: 生成无穷大, 1: 生成最大正常数
OSC [15] FP8转换指令溢出饱和 0: 生成无穷大或NaN, 1: 生成最大正常数

LSCALE, LSCALE2, NSCALE - 缩放控制

这些字段控制浮点数的缩放：

字段名 位范围 功能
LSCALE [22:16] 降缩放值（无符号整数），从结果指数中减去
LSCALE2 [37:32] 第二个FP8输入流的降缩放值
NSCALE [31:24] 缩放值（有符号整数），加到操作数指数上

如何访问FPMR寄存器？

在ARM汇编中，使用MRS和MSR指令来读写FPMR寄存器：

读取FPMR寄存器

MRS <Xt>, FPMR

写入FPMR寄存器

MSR FPMR, <Xt>

注意：访问FPMR需要满足特定条件，包括EL级别、特性使能等，否则会导致未定义异常或陷入(trap)。

实际应用示例

Linux内核中的使用

在Linux内核中，FPMR通常由浮点单元初始化代码配置：

// 设置FP8格式和缩放参数
void init_fp8_mode(void) {
    uint64_t fpmr_value = 0;
    
    // 设置输入格式为E4M3，输出格式为E5M2
    fpmr_value |= (0x1 << 0);   // F8S1 = 001 (E4M3)
    fpmr_value |= (0x1 << 3);   // F8S2 = 001 (E4M3) 
    fpmr_value |= (0x0 << 6);   // F8D = 000 (E5M2)
    
    // 设置溢出时生成最大正常数
    fpmr_value |= (0x1 << 14);  // OSM = 1
    fpmr_value |= (0x1 << 15);  // OSC = 1
    
    // 写入FPMR寄存器
    asm volatile("MSR FPMR, %0" : : "r" (fpmr_value));
}

ATF (ARM Trusted Firmware) 中的使用

在ATF中，可能在安全监控调用中配置FPMR：

// ATF中处理FP8配置的示例
uint64_t handle_fp8_config_smc(uint64_t x1, uint64_t x2) {
    // 检查调用者权限
    if (!is_caller_privileged()) {
        return SMC_UNK;
    }
    
    // 配置FPMR寄存器
    write_fpmr(x1);
    
    // 记录审计日志
    log_security_event("FPMR configured", x1);
    
    return SMC_OK;
}

OP-TEE中的使用

在OP-TEE安全环境中，FPMR可以用于安全浮点计算：

// OP-TEE中安全FP8计算的示例
TEE_Result secure_fp8_computation(void) {
    uint64_t original_fpmr;
    uint64_t secure_fpmr = 0;
    
    // 保存原始FPMR值
    asm volatile("MRS %0, FPMR" : "=r" (original_fpmr));
    
    // 配置安全计算所需的FPMR设置
    secure_fpmr |= (0x1 << 0);   // F8S1 = E4M3
    secure_fpmr |= (0x1 << 3);   // F8S2 = E4M3
    secure_fpmr |= (0x0 << 6);   // F8D = E5M2
    secure_fpmr |= (0x1 << 15);  // 溢出时饱和
    
    // 应用安全配置
    asm volatile("MSR FPMR, %0" : : "r" (secure_fpmr));
    
    // 执行安全敏感的FP8计算
    perform_secure_calculation();
    
    // 恢复原始FPMR设置
    asm volatile("MSR FPMR, %0" : : "r" (original_fpmr));
    
    return TEE_SUCCESS;
}

使用注意事项

2025@Arm精选课程，专注ARM架构和TEE安全，vx: sami01_2023

店铺二维码

店铺二维码

客服微信二维码

客服微信二维码