JEP 460: Vector API (Seventh Incubator) 总结

概述

JEP 460，即 Vector API（第七个孵化器版本），已在 JDK 22 中完成并交付。该 JEP 隶属于 Project Panama，旨在通过向量计算提升 Java 程序的性能。此版本在前六个孵化器版本（JDK 16 至 JDK 21）的基础上进行了增强，特别是对 JVM 编译器接口（JVMCI）的改进，以支持 Vector API 值。

主要目标

Vector API 的主要目标是：

1. 平台无关性：提供跨 CPU 架构的高效实现，包括 x64 和 AArch64。
2. 运行时编译与性能：确保向量操作能够可靠地映射到硬件指令，如 SSE、AVX（x64）和 NEON、SVE（AArch64）。
3. 优雅降级：在不支持某些向量指令的架构上，确保功能不受显著影响。

关键改进

1. 内存段支持：扩展了对堆内存段（MemorySegments）的访问支持，不再仅限于字节数组，而是支持任何基本元素类型的数组。
2. JVMCI 增强：改进了 JVM 编译器接口，以更好地支持 Vector API。

代码示例

传统标量加法与向量加法的对比：

标量加法

public void scalarAddition(int[] a, int[] b, int[] result) {
    for (int i = 0; i < a.length; i++) {
        result[i] = a[i] + b[i];
    }
}

向量加法

import jdk.incubator.vector.IntVector;
import jdk.incubator.vector.VectorSpecies;

public void vectorAddition(int[] a, int[] b, int[] result) {
    final VectorSpecies<Integer> species = IntVector.SPECIES_PREFERRED;
    int length = species.loopBound(a.length);
    for (int i = 0; i < length; i += species.length()) {
        IntVector va = IntVector.fromArray(species, a, i);
        IntVector vb = IntVector.fromArray(species, b, i);
        IntVector vc = va.add(vb);
        vc.intoArray(result, i);
    }
    // 处理剩余元素
    for (int i = length; i < a.length; i++) {
        result[i] = a[i] + b[i];
    }
}

向量加法通过 SIMD 指令并行处理多个元素，显著提升了性能。

长期目标

1. 与 Project Valhalla 对齐：计划将当前基于值的类改为值类，以进一步提升性能和内存效率。
2. 减少 API 表面积：在保持灵活性的同时，减少 API 的复杂性，以支持广泛的向量操作。

结论

Vector API 的持续孵化反映了 Java 社区对提升计算效率和开发者生产力的承诺。通过优化硬件指令的映射，Vector API 为 Java 在机器学习、加密和线性代数等领域的应用提供了显著的性能提升。