diff options
| author | y-jan137 <yousefjan24000@gmail.com> | 2026-03-14 14:49:46 +0300 |
|---|---|---|
| committer | y-jan137 <yousefjan24000@gmail.com> | 2026-03-14 14:49:46 +0300 |
| commit | 7db0d82731ababdf0b2f4d986dd54da3b4650954 (patch) | |
| tree | 145564ba45b9ae74a7eeebc319027c730c7177bc /src/matrix.cpp | |
| parent | 5ac9489c079f3f1a0ba1d2d8385001ada83a13b8 (diff) | |
Add triangular solve
Diffstat (limited to 'src/matrix.cpp')
| -rw-r--r-- | src/matrix.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/matrix.cpp b/src/matrix.cpp index 1faab25..c27264b 100644 --- a/src/matrix.cpp +++ b/src/matrix.cpp @@ -11,7 +11,7 @@ #include <immintrin.h> #endif -#if !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && !defined(__clangd__) +#if !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && defined(__ARM_FEATURE_FP64_VECTOR_ARITHMETIC) #include <arm_neon.h> #endif @@ -114,7 +114,7 @@ double dot_product_avx(const double* lhs, const double* rhs, std::size_t count) } #endif -#if !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && !defined(__clangd__) +#if !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && defined(__ARM_FEATURE_FP64_VECTOR_ARITHMETIC) double horizontal_sum(float64x2_t values) { return vgetq_lane_f64(values, 0) + vgetq_lane_f64(values, 1); } @@ -146,7 +146,7 @@ double dot_product_simd(const double* lhs, const double* rhs, std::size_t count) return dot_product_avx2(lhs, rhs, count); #elif !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__AVX__) && !defined(__AVX2__) return dot_product_avx(lhs, rhs, count); -#elif !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && !defined(__clangd__) +#elif !defined(LINEAR_ALGEBRA_FORCE_SCALAR_MATMUL) && defined(__ARM_NEON) && defined(__aarch64__) && defined(__ARM_FEATURE_FP64_VECTOR_ARITHMETIC) return dot_product_neon(lhs, rhs, count); #else return dot_product_scalar(lhs, rhs, count); |