Matrix Types¶
Clang provides a C/C++ language extension that allows users to directly express fixed-size 2-dimensional matrices as language values and perform arithmetic on them.
This feature is currently experimental, and both its design and its implementation are in flux.
Draft Specification¶
Matrix Type¶
A matrix type is a scalar type with an underlying element type, a constant
number of rows, and a constant number of columns. Matrix types with the same
element type, rows, and columns are the same type. A value of a matrix type
includes storage for rows * columns
values of the element type. The
internal layout, overall size and alignment are implementation-defined.
The maximum of the product of the number of rows and columns is implementation-defined. If that implementation-defined limit is exceeded, the program is ill-formed.
Currently, the element type of a matrix is only permitted to be one of the following types:
an integer type (as in C2x 6.2.5p19), but excluding enumerated types and
_Bool
the standard floating types
float
ordouble
a half-precision floating point type, if one is supported on the target
Other types may be supported in the future.
Matrix Type Attribute¶
Matrix types can be declared by adding the matrix_type
attribute to the
declaration of a typedef (or a C++ alias declaration). The underlying type
of the typedef must be a valid matrix element type. The
attribute takes two arguments, both of which must be integer constant
expressions that evaluate to a value greater than zero. The first specifies the
number of rows, and the second specifies the number of columns. The underlying
type of the typedef becomes a matrix type with the given dimensions and an
element type of the former underlying type.
If a declaration of a typedef-name has a matrix_type
attribute, then all
declaration of that typedef-name shall have a matrix_type attribute with the
same element type, number of rows, and number of columns.
Standard Conversions¶
The standard conversions are extended as follows. Note that these conversions are intentionally not listed as satisfying the constraints for assignment, which is to say, they are only permitted as explicit casts, not as implicit conversions.
A value of matrix type can be converted to another matrix type if the number of rows and columns are the same and the value’s elements can be converted to the element type of the result type. The result is a matrix where each element is the converted corresponding element.
A value of any real type (as in C2x 6.2.5p17) can be converted to a matrix type if it can be converted to the element type of the matrix. The result is a matrix where all elements are the converted original value.
If the number of rows or columns differ between the original and resulting type, the program is ill-formed.
Arithmetic Conversions¶
The usual arithmetic conversions are extended as follows.
Insert at the start:
If both operands are of matrix type, no arithmetic conversion is performed.
If one operand is of matrix type and the other operand is of a real type, convert the real type operand to the matrix type according to the standard conversion rules.
Matrix Type Element Access Operator¶
An expression of the form E1 [E2] [E3]
, where E1
has matrix type cv
M
, is a matrix element access expression. Let T
be the element type
of M
, and let R
and C
be the number of rows and columns in M
respectively. The index expressions shall have integral or unscoped
enumeration type and shall not be uses of the comma operator unless
parenthesized. The first index expression shall evaluate to a
non-negative value less than R
, and the second index expression shall
evaluate to a non-negative value less than C
, or else the expression has
undefined behavior. If E1
is a prvalue, the result is a prvalue with type
T
and is the value of the element at the given row and column in the matrix.
Otherwise, the result is a glvalue with type cv T
and with the same value
category as E1
which refers to the element at the given row and column in
the matrix.
Programs containing a single subscript expression into a matrix are ill-formed.
Note: We considered providing an expression of the form
postfix-expression [expression]
to access columns of a matrix. We think
that such an expression would be problematic once both column and row major
matrixes are supported: depending on the memory layout, either accessing columns
or rows can be done efficiently, but not both. Instead, we propose to provide
builtins to extract rows and columns from a matrix. This makes the operations
more explicit.
Matrix Type Binary Operators¶
Given two matrixes, the +
and -
operators perform element-wise addition
and subtraction, while the *
operator performs matrix multiplication.
+
, -
, *
, and /
can also be used with a matrix and a scalar
value, applying the operation to each element of the matrix.
Earlier versions of this extension did not support division by a scalar.
You can test for the availability of this feature with
__has_extension(matrix_types_scalar_division)
.
For the expression M1 BIN_OP M2
where
BIN_OP
is one of+
or-
, one ofM1
andM2
is of matrix type, and the other is of matrix type or real type; orBIN_OP
is*
, one ofM1
andM2
is of matrix type, and theother is of a real type; or
BIN_OP
is/
,M1
is of matrix type, andM2
is of a real type:The usual arithmetic conversions are applied to
M1
andM2
. [ Note: ifM1
orM2
are of a real type, they are broadcast to matrices here. — end note ]M1
andM2
shall be of the same matrix type.The result is equivalent to Res in the following where col is the number of columns and row is the number of rows in the matrix type:
decltype(M1) Res;
for (int C = 0; C < col; ++C)
for (int R = 0; R < row; ++R)
Res[R][C] = M1[R][C] BIN_OP M2[R][C];
Given the expression M1 * M2
where M1
and M2
are of matrix type:
The usual arithmetic conversions are applied to
M1
andM2
.The type of
M1
shall have the same number of columns as the type ofM2
has rows. The element types ofM1
andM2
shall be the same type.The resulting type,
MTy
, is a matrix type with the common element type, the number of rows ofM1
and the number of columns ofM2
.The result is equivalent to
Res
in the following whereEltTy
is the element type ofMTy
,col
is the number of columns,row
is the number of rows inMTy
andinner
is the number of columns ofM1
:
MTy Res;
for (int C = 0; C < col; ++C) {
for (int R = 0; R < row; ++R) {
EltTy Elt = 0;
for (int K = 0; K < inner; ++K) {
Elt += M1[R][K] * M2[K][C];
}
Res[R][C] = Elt;
}
All operations on matrix types match the behavior of the element type with respect to signed overflows.
With respect to floating-point contraction, rounding and environment rules, operations on matrix types match the behavior of the elementwise operations in the corresponding expansions provided above.
Operations on floating-point matrices have the same rounding and floating-point
environment behavior as ordinary floating-point operations in the expression’s
context. For the purposes of floating-point contraction, all calculations done
as part of a matrix operation are considered intermediate operations, and their
results need not be rounded to the format of the element type until the final
result in the containing expression. This is subject to the normal restrictions
on contraction, such as #pragma STDC FP_CONTRACT
.
For the +=
, -=
and *=
operators the semantics match their expanded
variants.
Matrix Type Builtin Operations¶
Each matrix type supports a collection of builtin expressions that look like function calls but do not form an overload set. Here they are described as function declarations with rules for how to construct the argument list types and return type and the library description elements from [library.description.structure.specifications]/3 in the C++ standard.
Definitions:
M, M1, M2, M3 - Matrix types
T - Element type
row, col - Row and column arguments respectively.
M2 __builtin_matrix_transpose(M1 matrix)
Remarks: The return type is a cv-unqualified matrix type that has the same
element type as M1
and has the same number of rows as M1
has columns and
the same number of columns as M1
has rows.
Returns: A matrix Res
equivalent to the code below, where col
refers to the
number of columns of M
, and row
to the number of rows of M
.
Effects: Equivalent to:
M Res;
for (int C = 0; C < col; ++C)
for (int R = 0; R < row; ++R)
Res[C][R] = matrix[R][C];
M __builtin_matrix_column_major_load(T *ptr, size_t row, size_t col, size_t columnStride)
Mandates: row
and col
shall be integral constants greater than 0.
Preconditions: columnStride
is greater than or equal to row
.
Remarks: The return type is a cv-unqualified matrix type with an element
type of the cv-unqualified version of T
and a number of rows and columns equal
to row
and col
respectively. The parameter columnStride
is optional
and if omitted row
is used as columnStride
.
Returns: A matrix Res
equivalent to:
M Res;
for (size_t C = 0; C < col; ++C) {
for (size_t R = 0; R < row; ++K)
Res[R][C] = ptr[R];
ptr += columnStride
}
void __builtin_matrix_column_major_store(M matrix, T *ptr, size_t columnStride)
Preconditions: columnStride
is greater than or equal to the number of rows in M
.
Remarks: The type T
is the const-unqualified version of the matrix
argument’s element type. The parameter columnStride
is optional and if
omitted, the number of rows of M
is used as columnStride
.
Effects: Equivalent to:
for (size_t C = 0; C < columns in M; ++C) {
for (size_t R = 0; R < rows in M; ++K)
ptr[R] = matrix[R][C];
ptr += columnStride
}
TODOs¶
TODO: Does it make sense to allow M::element_type, M::rows, and M::columns where M is a matrix type? We don’t support this anywhere else, but it’s convenient. The alternative is using template deduction to extract this information. Also add spelling for C.
Future Work: Initialization syntax.
Decisions for the Implementation in Clang¶
This section details decisions taken for the implementation in Clang and is not part of the draft specification.
The elements of a value of a matrix type are laid out in column-major order without padding.
We propose to provide a Clang option to override this behavior and allow contraction of those operations (e.g. -ffp-contract=matrix).
TODO: Specify how matrix values are passed to functions.