基于SIMD优化字节数组重复复制函数的实现与问题排查

需求与初始实现 #

需要实现一个字节数组重复填充函数（类似MemCpyReplicate），目标数组长度固定为源数组的整数倍。
最初采用Unsafe.CopyBlockUnaligned实现，但性能未达预期，因此尝试通过SIMD的Vector<T>接口进行优化，过程中遇到两个核心难题：

• 非对齐地址的SIMD读写处理
• 小于向量长度（当前为32字节）的字节模式复制异常

已实现的向量化变体 #

针对上述问题，实现了两个优化版本：

• Repeat_CopyBlock：在向量内部重复源模式，当模式长度超过向量长度时，回退使用CopyBlock批量复制
• Repeat_NoCopyBlock：先将源模式复制出一段超过向量长度的内容，再基于这段内容批量复制向量块

测试异常现象 #

测试发现：

• 当源模式大小在2-32字节区间时，两个变体的运行结果均异常，怀疑问题出在移动源窗口读取的逻辑上
• 当源模式大小超过32字节时，两个变体的结果均符合预期

性能测试数据 #

实现代码 #

Repeat_CopyBlock #

// 此处为Repeat_CopyBlock的具体实现代码
public static void Repeat_CopyBlock(byte[] source, byte[] destination)
{
    int sourceLength = source.Length;
    int destLength = destination.Length;
    int vectorSize = Vector<byte>.Count;

    if (sourceLength == 0 || destLength % sourceLength != 0)
        throw new ArgumentException();

    // 向量内重复模式逻辑
    // ...
}

Repeat_NoCopyBlock #

// 此处为Repeat_NoCopyBlock的具体实现代码
public static void Repeat_NoCopyBlock(byte[] source, byte[] destination)
{
    int sourceLength = source.Length;
    int destLength = destination.Length;
    int vectorSize = Vector<byte>.Count;

    if (sourceLength == 0 || destLength % sourceLength != 0)
        throw new ArgumentException();

    // 先复制超向量长度的内容
    // ...
    // 批量复制向量块
    // ...
}

问题排查方向 #

针对小模式（2-32字节）的异常，核心排查点：

1. 向量内模式填充逻辑：当模式长度小于向量长度时，是否正确循环填充向量，而非错误地移动源数组窗口读取（会导致越界或读取无效数据）
2. 非对齐地址处理：是否使用Vector.LoadUnaligned/Vector.StoreUnaligned处理非对齐的源/目标地址，避免SIMD操作的内存访问错误
3. 边界处理：在批量复制向量块后，是否正确处理剩余的不足一个向量长度的字节（虽然目标长度是源的整数倍，但需确保最后一段的填充逻辑正确）

内容的提问来源于stack exchange，提问作者ListigerLurch