如何在Rust中对std::simd向量执行重解释转换？

该功能是否已存在？ #

目前Rust标准库的std::simd（对应core::simd）并未提供直接的跨元素类型、相同总大小的Simd类型重解释转换函数。标准库的Simd API偏向强类型设计，默认不鼓励随意的类型重解释，但可以通过安全且高效的方式手动实现。

正确的实现方式 #

你提供的字节转换实现虽然可行，但会涉及内存拷贝，性能不如直接的位级转换。更高效的方式是利用Simd::to_bits和Simd::from_bits方法（Rust 1.61+可用），它们直接操作Simd类型的位模式，无需内存拷贝：

use core::simd::{Simd, SimdElement, LaneCount, SupportedLaneCount};

pub fn reinterpret_cast<T1: SimdElement, T2: SimdElement, const N1: usize, const N2: usize>(
    x: Simd<T1, N1>,
) -> Simd<T2, N2>
where
    LaneCount<N1>: SupportedLaneCount,
    LaneCount<N2>: SupportedLaneCount,
{
    // 编译期断言确保两个Simd类型总大小一致
    const_assert_eq!(core::mem::size_of::<Simd<T1, N1>>(), core::mem::size_of::<Simd<T2, N2>>());
    
    // 通过位模式直接完成零开销转换
    Simd::from_bits(x.to_bits())
}

关键说明 #

• const_assert_eq!用于在编译阶段拦截两个Simd类型大小不匹配的情况，避免运行时未定义行为。
• to_bits/from_bits直接在寄存器层面操作位模式，完全没有内存拷贝开销，是性能最优的实现方式。

如果你的Rust版本不支持to_bits/from_bits，也可以使用core::mem::transmute实现，需注意类型安全约束：

use core::mem;
use core::simd::{Simd, SimdElement, LaneCount, SupportedLaneCount};

pub fn reinterpret_cast<T1: SimdElement, T2: SimdElement, const N1: usize, const N2: usize>(
    x: Simd<T1, N1>,
) -> Simd<T2, N2>
where
    LaneCount<N1>: SupportedLaneCount,
    LaneCount<N2>: SupportedLaneCount,
{
    const_assert_eq!(mem::size_of::<Simd<T1, N1>>(), mem::size_of::<Simd<T2, N2>>());
    // 安全transmute：大小一致且Simd类型内存布局兼容
    unsafe { mem::transmute(x) }
}

注意事项 #

• 无论哪种实现，必须保证两个Simd类型的总内存大小完全相同，否则会触发未定义行为。
• 位级转换不处理字节序，如果你需要强制小端/大端字节序的转换，你最初的字节转换方式依然适用，但性能会略低于位级转换。

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