mirror of
https://github.com/recp/cglm.git
synced 2025-10-04 09:08:53 +00:00
neon: implement matrix multiplication for ARM Neon
This commit is contained in:
3
.gitignore
vendored
3
.gitignore
vendored
@@ -54,3 +54,6 @@ test/test_*
|
|||||||
test-*
|
test-*
|
||||||
test/.libs/*
|
test/.libs/*
|
||||||
test/tests
|
test/tests
|
||||||
|
cglm_arm/*
|
||||||
|
cglm_test_ios/*
|
||||||
|
cglm_test_iosTests/*
|
57
include/arch/simd/neon/mat4.h
Normal file
57
include/arch/simd/neon/mat4.h
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c), Recep Aslantas.
|
||||||
|
*
|
||||||
|
* MIT License (MIT), http://opensource.org/licenses/MIT
|
||||||
|
* Full license can be found in the LICENSE file
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef cglm_mat4_neon_h
|
||||||
|
#define cglm_mat4_neon_h
|
||||||
|
#if defined(__ARM_NEON_FP)
|
||||||
|
|
||||||
|
#include "../../../cglm-common.h"
|
||||||
|
#include <arm_neon.h>
|
||||||
|
|
||||||
|
CGLM_INLINE
|
||||||
|
void
|
||||||
|
glm_mat4_mul_neon(mat4 m1, mat4 m2, mat4 dest) {
|
||||||
|
/* D = R * L (Column-Major) */
|
||||||
|
float32x4_t l0, l1, l2, l3, r, d0, d1, d2, d3;
|
||||||
|
|
||||||
|
l0 = vld1q_f32(m2[0]);
|
||||||
|
l1 = vld1q_f32(m2[1]);
|
||||||
|
l2 = vld1q_f32(m2[2]);
|
||||||
|
l3 = vld1q_f32(m2[3]);
|
||||||
|
|
||||||
|
r = vld1q_f32(m1[0]);
|
||||||
|
d0 = vmulq_lane_f32(r, vget_low_f32(l0), 0);
|
||||||
|
d1 = vmulq_lane_f32(r, vget_low_f32(l1), 0);
|
||||||
|
d2 = vmulq_lane_f32(r, vget_low_f32(l2), 0);
|
||||||
|
d3 = vmulq_lane_f32(r, vget_low_f32(l3), 0);
|
||||||
|
|
||||||
|
r = vld1q_f32(m1[1]);
|
||||||
|
d0 = vmlaq_lane_f32(d0, r, vget_low_f32(l0), 1);
|
||||||
|
d1 = vmlaq_lane_f32(d1, r, vget_low_f32(l1), 1);
|
||||||
|
d2 = vmlaq_lane_f32(d2, r, vget_low_f32(l2), 1);
|
||||||
|
d3 = vmlaq_lane_f32(d3, r, vget_low_f32(l3), 1);
|
||||||
|
|
||||||
|
r = vld1q_f32(m1[2]);
|
||||||
|
d0 = vmlaq_lane_f32(d0, r, vget_high_f32(l0), 0);
|
||||||
|
d1 = vmlaq_lane_f32(d1, r, vget_high_f32(l1), 0);
|
||||||
|
d2 = vmlaq_lane_f32(d2, r, vget_high_f32(l2), 0);
|
||||||
|
d3 = vmlaq_lane_f32(d3, r, vget_high_f32(l3), 0);
|
||||||
|
|
||||||
|
r = vld1q_f32(m1[3]);
|
||||||
|
d0 = vmlaq_lane_f32(d0, r, vget_high_f32(l0), 1);
|
||||||
|
d1 = vmlaq_lane_f32(d1, r, vget_high_f32(l1), 1);
|
||||||
|
d2 = vmlaq_lane_f32(d2, r, vget_high_f32(l2), 1);
|
||||||
|
d3 = vmlaq_lane_f32(d3, r, vget_high_f32(l3), 1);
|
||||||
|
|
||||||
|
vst1q_f32(dest[0], d0);
|
||||||
|
vst1q_f32(dest[1], d1);
|
||||||
|
vst1q_f32(dest[2], d2);
|
||||||
|
vst1q_f32(dest[3], d3);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
#endif /* cglm_mat4_neon_h */
|
@@ -45,8 +45,18 @@
|
|||||||
#define cglm_mat_h
|
#define cglm_mat_h
|
||||||
|
|
||||||
#include "cglm-common.h"
|
#include "cglm-common.h"
|
||||||
#include "arch/simd/cglm-mat-simd-sse2.h"
|
|
||||||
#include "arch/simd/cglm-mat-simd-avx.h"
|
#ifdef CGLM_SSE_FP
|
||||||
|
# include "arch/simd/cglm-mat-simd-sse2.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef CGLM_AVX_FP
|
||||||
|
# include "arch/simd/cglm-mat-simd-avx.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef CGLM_NEON_FP
|
||||||
|
# include "arch/simd/neon/mat4.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
|
||||||
@@ -215,6 +225,8 @@ glm_mat4_mul(mat4 m1, mat4 m2, mat4 dest) {
|
|||||||
glm_mat4_mul_avx(m1, m2, dest);
|
glm_mat4_mul_avx(m1, m2, dest);
|
||||||
#elif defined( __SSE__ ) || defined( __SSE2__ )
|
#elif defined( __SSE__ ) || defined( __SSE2__ )
|
||||||
glm_mat4_mul_sse2(m1, m2, dest);
|
glm_mat4_mul_sse2(m1, m2, dest);
|
||||||
|
#elif defined( __ARM_NEON_FP )
|
||||||
|
glm_mat4_mul_neon(m1, m2, dest);
|
||||||
#else
|
#else
|
||||||
float a00 = m1[0][0], a01 = m1[0][1], a02 = m1[0][2], a03 = m1[0][3],
|
float a00 = m1[0][0], a01 = m1[0][1], a02 = m1[0][2], a03 = m1[0][3],
|
||||||
a10 = m1[1][0], a11 = m1[1][1], a12 = m1[1][2], a13 = m1[1][3],
|
a10 = m1[1][0], a11 = m1[1][1], a12 = m1[1][2], a13 = m1[1][3],
|
||||||
|
Reference in New Issue
Block a user