Feature/alignment 8 (#141)

* Set RTNEURAL_DEFAULT_ALIGNMENT=8 on armv7 * Add Eigen::Aligned8 to RTNeuralEigenAlignment options
jatinchowdhury18 · Jun 22, 2024 · 8c1847d · 8c1847d
1 parent 680f620
commit 8c1847d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/RTNeural/common.h b/RTNeural/common.h
@@ -37,9 +37,13 @@ constexpr T ceil_div(T num, T den)
 namespace RTNEURAL_NAMESPACE
 {
 #if RTNEURAL_DEFAULT_ALIGNMENT == 32
-constexpr auto RTNeuralEigenAlignment = Eigen::Aligned32;
+    constexpr auto RTNeuralEigenAlignment = Eigen::Aligned32;
+#elif RTNEURAL_DEFAULT_ALIGNMENT == 16
+    constexpr auto RTNeuralEigenAlignment = Eigen::Aligned16;
+#elif RTNEURAL_DEFAULT_ALIGNMENT == 8
+    constexpr auto RTNeuralEigenAlignment = Eigen::Aligned8;
 #else
-constexpr auto RTNeuralEigenAlignment = Eigen::Aligned16;
+    #error "Unsupported alignment"
 #endif
 } // namespace RTNEURAL_NAMESPACE
 

diff --git a/cmake/SIMDExtensions.cmake b/cmake/SIMDExtensions.cmake
@@ -1,6 +1,10 @@
 option(RTNEURAL_USE_AVX "Enables AVX SIMD Support" OFF)
 if(NOT RTNEURAL_USE_AVX)
-    target_compile_definitions(RTNeural PUBLIC RTNEURAL_DEFAULT_ALIGNMENT=16)
+    if(CMAKE_SYSTEM_PROCESSOR MATCHES "armv7")
+        target_compile_definitions(RTNeural PUBLIC RTNEURAL_DEFAULT_ALIGNMENT=8)
+    else()
+        target_compile_definitions(RTNeural PUBLIC RTNEURAL_DEFAULT_ALIGNMENT=16)
+    endif()
 else()
     message(STATUS "RTNeural -- Attempting to enable AVX...")