6.52 Использование векторных инструкций через встроенные функции

В некоторых целевых платформах набор инструкций содержит векторные SIMD-инструкции, которые работают с несколькими значениями, содержащимися в одном большом регистре одновременно. Например, на x86 можно использовать расширения MMX, 3DNow! и SSE таким образом.

Первый шаг в использовании этих расширений — предоставление необходимых типов данных. Это следует делать с помощью соответствующего typedef:

typedef int v4si __attribute__ ((vector_size (16)));

Тип int указывает базовый тип, а атрибут указывает размер вектора для переменной, измеряемый в байтах. Например, приведенное выше объявление заставляет компилятор установить режим для типа v4si в ширину 16 байт и разделить на единицы размера int. Для 32-битного int это означает вектор из 4 единиц по 4 байта, а соответствующий режим foo — V4SI.

Атрибут vector_size применим только к целочисленным и плавающим скалярам, хотя массивы, указатели и значения возвращаемых функций разрешены в сочетании с этим конструктом. В настоящее время разрешены только размеры, которые являются положительными степенями двойки, кратными размеру базового типа.

Все основные целочисленные типы могут использоваться в качестве базовых типов, как со знаком, так и без знака: char, short, int, long, long long. Кроме того, float и double могут использоваться для построения типов векторных чисел с плавающей точкой.

Указание комбинации, которая не является допустимой для текущей архитектуры, приводит к тому, что GCC синтезирует инструкции с помощью более узкого режима. Например, если вы укажете переменную типа V4SI и ваша архитектура не допускает этот конкретный тип SIMD, GCC генерирует код, использующий 4 SIs.

Определенные таким образом типы можно использовать с подмножеством обычных операций C. В настоящее время GCC разрешает использование следующих операторов для этих типов: +, -, *, /, unary minus, ^, |, &, ~, %.

Операции ведут себя как операции C++ valarrays. Сложение определяется как сложение соответствующих элементов операндов. Например, в приведенном ниже коде каждый из 4 элементов в a складывается с соответствующими 4 элементами в b, а полученный вектор сохраняется в c.

typedef int v4si __attribute__ ((vector_size (16)));

v4si a, b, c;

c = a + b;

Вычитание, умножение, деление и логические операции работают аналогичным образом. Аналогичным образом, результат применения унарных операторов минус или дополнения к типу вектора — это вектор, чьи элементы являются отрицательными или дополненными значениями соответствующих элементов в операнде.

Можно использовать операторы сдвига <<, >> для векторов целочисленного типа. Операция определяется следующим образом: {a0, a1, …, an} >> {b0, b1, …, bn} == {a0 >> b0, a1 >> b1, …, an >> bn}. Векторные операнды должны иметь одинаковое количество элементов.

Для удобства допускается использование бинарной векторной операции, где один операнд является скаляром. В этом случае компилятор преобразует скалярный операнд в вектор, где каждый элемент — это скаляр из операции. Преобразование происходит только в том случае, если скаляр может быть безопасно преобразован в тип элемента вектора. Рассмотрим следующий код.

typedef int v4si __attribute__ ((vector_size (16)));

v4si a, b, c;
long l;

a = b + 1;    /* a = b + {1,1,1,1}; */
a = 2 * b;    /* a = {2,2,2,2} * b; */

a = l + a;    /* Error, cannot convert long to int. */

Векторы можно индексировать, как если бы вектор был массивом с тем же количеством элементов и базовым типом. Доступ за пределы границ вызывает неопределенное поведение во время выполнения. Предупреждения о доступе за пределы границ для индексирования векторов можно включить с помощью -Warray-bounds.

Поддержка сравнения векторов с помощью стандартных операторов сравнения: ==, !=, <, <=, >, >=. Операнды сравнения могут быть векторными выражениями целочисленного или вещественного типа. Сравнение векторов целочисленного типа и векторов вещественного типа не поддерживается. Результатом сравнения является вектор той же ширины и количества элементов, что и операнды сравнения, с целочисленным элементом типа со знаком.

Векторы сравниваются по элементам, выдавая 0, когда сравнение ложно, и -1 (константа соответствующего типа, где все биты установлены) в противном случае. Рассмотрим следующий пример.

typedef int v4si __attribute__ ((vector_size (16)));

v4si a = {1,2,3,4};
v4si b = {3,2,1,4};
v4si c;

c = a >  b;     /* The result would be {0, 0,-1, 0}  */
c = a == b;     /* The result would be {0,-1, 0,-1}  */

В C++ доступен тернарный оператор ?:. a?b:c, где b и c — векторы одного типа, а a — целочисленный вектор с тем же количеством элементов того же размера, что и b и c, вычисляет все три аргумента и создает вектор {a[0]?b[0]:c[0], a[1]?b[1]:c[1], …}. Обратите внимание, что в отличие от OpenCL, a интерпретируется как a != 0, а не как a < 0. Как и в случае с бинарными операциями, этот синтаксис также допускается, когда один из b или c является скаляром, который затем преобразуется в вектор. Если оба b и c являются скалярами, а тип true?b:c имеет тот же размер, что и тип элемента a, то b и c преобразуются в тип вектора, элементы которого имеют этот тип и имеют то же количество элементов, что и a.

В C++ доступны логические операторы !, &&, || для векторов. !v эквивалентно v == 0, a && b эквивалентно a!=0 & b!=0, а a || b эквивалентно a!=0 | b!=0. При смешанных операциях между скаляром s и вектором v, s && v эквивалентно s?v!=0:0 (оценка — по короткому замыканию), а v && s эквивалентно v!=0 & (s?-1:0).

Векторное перемешивание доступно с помощью функций __builtin_shuffle (vec, mask) и __builtin_shuffle (vec0, vec1, mask). Обе функции строят перестановку элементов из одного или двух векторов и возвращают вектор того же типа, что и входной(е) вектор(ы). mask — это целочисленный вектор с той же шириной (W) и количеством элементов (N), что и выходной вектор.

Элементы входных векторов нумеруются в порядке памяти vec0, начиная с 0, и vec1, начиная с N. Элементы mask рассматриваются как остаток от деления на N в случае с одним операндом и как остаток от деления на 2*N в случае с двумя операндами.

Рассмотрим следующий пример:

typedef int v4si __attribute__ ((vector_size (16)));

v4si a = {1,2,3,4};
v4si b = {5,6,7,8};
v4si mask1 = {0,1,1,3};
v4si mask2 = {0,4,2,5};
v4si res;

res = __builtin_shuffle (a, mask1);       /* res is {1,2,2,4}  */
res = __builtin_shuffle (a, b, mask2);    /* res is {1,5,3,6}  */

Обратите внимание, что __builtin_shuffle намеренно семантически совместимо с функциями OpenCL shuffle и shuffle2.

Вы можете объявлять переменные и использовать их в вызовах функций и возвращаемых значениях, а также в присваиваниях и некоторых преобразованиях. Вы можете указать тип вектора как возвращаемый тип функции. Типы векторов также могут использоваться в качестве аргументов функций. Преобразование от одного типа вектора к другому возможно при условии, что они имеют одинаковый размер (фактически, вы также можете преобразовывать векторы в другие типы данных того же размера).

Вы не можете выполнять операции между векторами различной длины или различной знакомости без преобразования.

Векторное перемешивание доступно с использованием функции __builtin_shufflevector (vec1, vec2, index...).

vec1 и vec2 должны быть выражениями с векторным типом и совместимым типом элемента. Результатом __builtin_shufflevector является вектор с тем же типом элемента, что и vec1 и vec2, но с количеством элементов, равным количеству указанных индексов.

Аргументы index — это список целых чисел, которые указывают индексы элементов первых двух векторов, которые должны быть извлечены и возвращены в новом векторе. Эти индексы элементов нумеруются последовательно, начиная с первого вектора и продолжаясь во втором векторе. Индекс -1 может использоваться для указания того, что соответствующий элемент в возвращаемом векторе — это «неважно» и может быть свободно выбран для оптимизации генерируемой последовательности кода, выполняющей операцию перемешивания.

Рассмотрим следующий пример:

typedef int v4si __attribute__ ((vector_size (16)));
typedef int v8si __attribute__ ((vector_size (32)));

v8si a = {1,-2,3,-4,5,-6,7,-8};
v4si b = __builtin_shufflevector (a, a, 0, 2, 4, 6); /* b is {1,3,5,7} */
v4si c = {-2,-4,-6,-8};
v8si d = __builtin_shufflevector (c, b, 4, 0, 5, 1, 6, 2, 7, 3); /* d is a */

Преобразование векторов доступно с помощью функции __builtin_convertvector (vec, vectype).

vec должно быть выражением с целочисленным или вещественным векторным типом, а vectype — целочисленным или вещественным векторным типом с тем же количеством элементов. Результат имеет тип vectype и значение C-преобразования каждого элемента vec в тип элемента vectype.

Рассмотрим следующий пример:

typedef int v4si __attribute__ ((vector_size (16)));
typedef float v4sf __attribute__ ((vector_size (16)));
typedef double v4df __attribute__ ((vector_size (32)));
typedef unsigned long long v4di __attribute__ ((vector_size (32)));

v4si a = {1,-2,3,-4};
v4sf b = {1.5f,-2.5f,3.f,7.f};
v4di c = {1ULL,5ULL,0ULL,10ULL};
v4sf d = __builtin_convertvector (a, v4sf); /* d is {1.f,-2.f,3.f,-4.f} */
/* Equivalent of:
   v4sf d = { (float)a[0], (float)a[1], (float)a[2], (float)a[3] }; */
v4df e = __builtin_convertvector (a, v4df); /* e is {1.,-2.,3.,-4.} */
v4df f = __builtin_convertvector (b, v4df); /* f is {1.5,-2.5,3.,7.} */
v4si g = __builtin_convertvector (f, v4si); /* g is {1,-2,3,7} */
v4si h = __builtin_convertvector (c, v4si); /* h is {1,5,0,10} */

Иногда желательно писать код, смешивая общие векторные операции (для ясности) и машиноспецифические векторные встроенные функции (для доступа к векторным инструкциям, которые не предоставляются общими встроенными функциями). На x86 встроенные функции для целочисленных векторов обычно используют тот же тип вектора __m128i независимо от того, как они интерпретируют вектор, что делает необходимым преобразовывать их аргументы и возвращаемые значения из/в другие типы векторов. В C вы можете использовать тип union:

#include <immintrin.h>

typedef unsigned char u8x16 __attribute__ ((vector_size (16)));
typedef unsigned int  u32x4 __attribute__ ((vector_size (16)));

typedef union {
        __m128i mm;
        u8x16   u8;
        u32x4   u32;
} v128;

для переменных, которые могут использоваться как с встроенными операторами, так и с x86 встроенными функциями:

v128 x, y = { 0 };
memcpy (&x, ptr, sizeof x);
y.u8  += 0x80;
x.mm  = _mm_adds_epu8 (x.mm, y.mm);
x.u32 &= 0xffffff;

/* Instead of a variable, a compound literal may be used to pass the
   return value of an intrinsic call to a function expecting the union: */
v128 foo (v128);
x = foo ((v128) {_mm_adds_epu8 (x.mm, y.mm)});

Далее: Offsetof, Предыдущий: Возврат адреса, Вверх: Расширения C [Оглавление][Индекс]