如何将Eigen矩阵与向量类型转换为Apple simd类型以传入GPU？

我在应用代码中使用Eigen的向量和矩阵类型（如Vector3f、Matrix3f）处理图形相关线性代数运算，但需要通过Apple的Metal API将其传入GPU。Metal使用<simd/simd.h>头文件中的类型，这些类型的声明如下：

// From <simd/vector_types.h>
typedef __attribute__((__ext_vector_type__(3))) float simd_float3;
...
// From <simd/types.h>
typedef struct { simd_float3 columns[3]; } simd_float3x3;

请问将Eigen的Vector3f转换为simd_float3、Matrix3f转换为simd_float3x3的最高效方式是什么？或许二者内存布局一致无需转换？若如此最好能用static_assert让编译器验证。

补充说明：我假设无法在Metal着色器代码中使用Eigen类型，只能使用simd类型（如simd_float3、simd_float3x3等）。若能无性能损失或无重大不便地直接使用Eigen类型，我也会考虑这样做。

1. 内存布局一致性验证 #

Eigen与Metal simd类型的内存布局完全匹配，可以通过编译期断言验证，避免运行时错误：

对于Vector3f和simd_float3

Eigen::Vector3f是连续存储的3个float，simd_float3作为扩展向量类型，同样是连续的3个float，二者大小、对齐要求一致：

#include <Eigen/Core>
#include <simd/simd.h>

static_assert(sizeof(Eigen::Vector3f) == sizeof(simd_float3), 
              "Vector size mismatch between Eigen and Metal simd types");
static_assert(alignof(Eigen::Vector3f) == alignof(simd_float3), 
              "Vector alignment mismatch between Eigen and Metal simd types");

对于Matrix3f和simd_float3x3

Eigen默认采用列主序存储矩阵，Matrix3f的内存布局是连续的9个float，按列排列；simd_float3x3是3个simd_float3列向量的数组，同样是列主序的连续9个float，二者布局完全匹配：

static_assert(sizeof(Eigen::Matrix3f) == sizeof(simd_float3x3), 
              "Matrix size mismatch between Eigen and Metal simd types");
static_assert(alignof(Eigen::Matrix3f) == alignof(simd_float3x3), 
              "Matrix alignment mismatch between Eigen and Metal simd types");
// 确保Eigen矩阵是列主序（默认即为列主序，此断言用于防止意外修改存储顺序）
static_assert(Eigen::Matrix3f::IsColMajor, 
              "Eigen matrix must be column-major to match Metal simd matrix layout");

2. 零开销转换方式 #

由于内存布局完全一致，无需拷贝数据，直接通过类型转换即可完成，是最高效的方式：

C++17及更早版本：reinterpret_cast

// Vector3f → simd_float3
Eigen::Vector3f eigen_vec(1.0f, 2.0f, 3.0f);
simd_float3 simd_vec = *reinterpret_cast<simd_float3*>(&eigen_vec);

// Matrix3f → simd_float3x3
Eigen::Matrix3f eigen_mat;
eigen_mat << 1, 2, 3,
             4, 5, 6,
             7, 8, 9;
simd_float3x3 simd_mat = *reinterpret_cast<simd_float3x3*>(&eigen_mat);

C++20及以上：std::bit_cast（更安全，编译期检查类型大小匹配）

#include <bit>

// Vector3f → simd_float3
simd_float3 simd_vec = std::bit_cast<simd_float3>(eigen_vec);

// Matrix3f → simd_float3x3
simd_float3x3 simd_mat = std::bit_cast<simd_float3x3>(eigen_mat);

3. 关于Metal着色器中使用Eigen的说明 #

Metal着色器采用MSL（Metal Shading Language），Eigen是C++库，无法直接在MSL中使用，因此必须将Eigen类型转换为Metal原生的simd类型后再传入GPU，没有无损失直接使用的途径。

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