From 983a3a52bd62eca5da3b3a3797b50af4f4ca3787 Mon Sep 17 00:00:00 2001
From: Konstantinos Margaritis <konstantinos@vectorcamp.gr>
Date: Sat, 7 Oct 2023 22:27:26 +0800
Subject: [PATCH] include extra sources for Arm on non-fat builds

---
 CMakeLists.txt | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index bde8b7e6..ad7ed180 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -820,6 +820,8 @@ set (hs_exec_neon_SRCS
     src/nfa/vermicelli_simd.cpp)
 set (hs_exec_sve_SRCS
     src/nfa/vermicelli_simd.cpp)
+set (hs_exec_sve2_SRCS
+    src/nfa/vermicelli_simd.cpp)
 
 set (hs_exec_avx2_SRCS
     src/fdr/teddy_avx2.c
@@ -1278,8 +1280,14 @@ if (NOT FAT_RUNTIME)
         set(hs_exec_SRCS ${hs_exec_SRCS} ${hs_exec_avx2_SRCS})
     endif()
 
-    if (NOT BUILD_SVE2)
-        set(hs_exec_SRCS ${hs_exec_SRCS} ${hs_exec_neon_SRCS})
+    if (ARCH_AARCH64)
+        if (BUILD_SVE2)
+            set(hs_exec_SRCS ${hs_exec_SRCS} ${hs_exec_sve2_SRCS})
+        elseif (BUILD_SVE)
+            set(hs_exec_SRCS ${hs_exec_SRCS} ${hs_exec_sve_SRCS})
+        else()
+            set(hs_exec_SRCS ${hs_exec_SRCS} ${hs_exec_neon_SRCS})
+        endif()
     endif()
 
     if (BUILD_STATIC_LIBS)