LLVM中矩阵Matrix的实现分析

1 背景说明

Clang提供了C/C++语言对矩阵的扩展支持，以方便用户使用可变大小的二维数据类型来实现计算，目前该特性还是实验版，设计和实现都在变化中。LLVM目前设计为支持小型列矩阵（column major），其对矩阵的设计基于向量。为用户提供向量代码生成功能，能够减少不必要的内存访问，并且提供用户友好的接口。本文基于LLVM13以Intel X86架构为平台，来实验和分析LLVM编译器对矩阵运算的基础支持。
LLVM目前支持的矩阵间运算包括矩阵转置、矩阵加减法和矩阵乘法，矩阵除法只支持矩阵除以标量，不支持矩阵与矩阵相除。

2 功能实现

2.1 数据实现

矩阵支持的元素类型有整型、单精、双精、半精，用户可以通过matrix_type属性来定义各种不同元素类型不同行列数的矩阵。

2.2 内建支持

2.2.1 Clang

Clang前端提供了3个内建接口供用户使用，如下表所示，目前矩阵的load和store都是按列操作，未来对行矩阵（row major）的支持也在计划中。

接口名称	功能	参数	返回值
__builtin_matrix_column_major_load	按列加载矩阵	T *ptr：加载起始地址 size_t row：矩阵行数 size_t col：矩阵列数 size_t columnStride：跨步，指加载完一列数据后ptr自增的元素个数	row行col列T元素类型矩阵
__builtin_matrix_column_major_store	按列存储矩阵	dx4x4_t matrix：待存储的矩阵 T *ptr：存储起始地址 size_t columnStride：跨步，指ptr存储完一列数据后自增的元素个数	无
__builtin_matrix_transpose	矩阵转置	dx4x4_t matrix：待转置的矩阵	转置后的矩阵

当从一个基地址*ptr加载矩阵时，列矩阵（column major）是一列一列依次赋值的。下图展示了跨步为5时， 列矩阵（column major）的load和store操作。

2.2.1 LLVM

LLVM后端提供了int_matrix_multiply内建函数，该接口并未在Clang前端暴露，用户在编码矩阵乘计算时，只需要使用乘法符号（*），LLVM在创建中间表示（IR）的时候会自动创建对该内建函数的调用。

2.3 算法流程

通过clang -cc1 -fenable-matrix -emit-llvm matrix_load.c编译代码，生成的IR中包含对内建接口的调用。