diff --git a/include/cglm-affine-mat-avx.h b/include/cglm-affine-mat-avx.h new file mode 100644 index 0000000..096fd15 --- /dev/null +++ b/include/cglm-affine-mat-avx.h @@ -0,0 +1,61 @@ +/* + * Copyright (c), Recep Aslantas. + * + * MIT License (MIT), http://opensource.org/licenses/MIT + * Full license can be found in the LICENSE file + */ + +#ifndef cglm_affine_mat_avx_h +#define cglm_affine_mat_avx_h +#ifdef __AVX__ + +#include "cglm-intrin.h" +#include + +CGLM_INLINE +void +glm_affine_mul_avx(mat4 m1, mat4 m2, mat4 dest) { + /* D = R * L (Column-Major) */ + + __m256 y0, y1, y2, y3, y4, y5, y6, y7, y8, y9; + + y0 = _mm256_load_ps(m2[0]); /* h g f e d c b a */ + y1 = _mm256_load_ps(m2[2]); /* p o n m l k j i */ + + y2 = _mm256_load_ps(m1[0]); /* h g f e d c b a */ + y3 = _mm256_load_ps(m1[2]); /* p o n m l k j i */ + + y4 = _mm256_permute2f128_ps(y2, y2, 0b00000011); /* d c b a h g f e */ + y5 = _mm256_permute2f128_ps(y3, y3, 0b00000000); /* l k j i l k j i */ + + /* f f f f a a a a */ + /* g g g g c c c c */ + /* e e e e b b b b */ + y7 = _mm256_permute_ps(y0, 0b10101010); + y6 = _mm256_permutevar_ps(y0, _mm256_set_epi32(1, 1, 1, 1, 0, 0, 0, 0)); + y8 = _mm256_permutevar_ps(y0, _mm256_set_epi32(0, 0, 0, 0, 1, 1, 1, 1)); + + _mm256_store_ps(dest[0], + _mm256_add_ps(_mm256_add_ps(_mm256_mul_ps(y2, y6), + _mm256_mul_ps(y4, y8)), + _mm256_mul_ps(y5, y7))); + + + /* n n n n i i i i */ + /* p p p p k k k k */ + /* m m m m j j j j */ + /* o o o o l l l l */ + y6 = _mm256_permutevar_ps(y1, _mm256_set_epi32(1, 1, 1, 1, 0, 0, 0, 0)); + y7 = _mm256_permutevar_ps(y1, _mm256_set_epi32(3, 3, 3, 3, 2, 2, 2, 2)); + y8 = _mm256_permutevar_ps(y1, _mm256_set_epi32(0, 0, 0, 0, 1, 1, 1, 1)); + y9 = _mm256_permutevar_ps(y1, _mm256_set_epi32(2, 2, 2, 2, 3, 3, 3, 3)); + + _mm256_store_ps(dest[2], + _mm256_add_ps(_mm256_add_ps(_mm256_mul_ps(y2, y6), + _mm256_mul_ps(y3, y7)), + _mm256_add_ps(_mm256_mul_ps(y4, y8), + _mm256_mul_ps(y5, y9)))); +} + +#endif +#endif /* cglm_affine_mat_avx_h */ diff --git a/include/cglm-affine-mat.h b/include/cglm-affine-mat.h index 78d9542..adb9f0d 100644 --- a/include/cglm-affine-mat.h +++ b/include/cglm-affine-mat.h @@ -11,12 +11,15 @@ #include "cglm.h" #include "cglm-mat.h" #include "cglm-affine-mat-sse2.h" +#include "cglm-affine-mat-avx.h" #include CGLM_INLINE void glm_affine_mul(mat4 m1, mat4 m2, mat4 dest) { -#if defined( __SSE__ ) || defined( __SSE2__ ) +#ifdef __AVX__ + glm_affine_mul_avx(m1, m2, dest); +#elif defined( __SSE__ ) || defined( __SSE2__ ) glm_affine_mul_sse2(m1, m2, dest); #else float a00, a01, a02, a03, b00, b01, b02,