zlib-ng-2.2.1-e2k.patch

From 02db39f2b1ea78859118c177078cee20552ec388 Mon Sep 17 00:00:00 2001
From: Ilya Kurdyukov <jpegqs@gmail.com>
Date: Sat, 6 Jul 2024 20:54:48 +0700
Subject: [PATCH] zlib-ng-2.2.1 e2k support

---
 CMakeLists.txt                | 14 +++++++++++++-
 arch/x86/x86_features.c       | 14 ++++++++++++++
 arch/x86/x86_intrins.h        |  7 ++++++-
 cmake/detect-arch.cmake       |  2 ++
 cmake/detect-intrinsics.cmake |  3 +++
 5 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 719673f..03c33da 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -115,6 +115,18 @@ elseif(BASEARCH_S360_FOUND)
     option(WITH_DFLTCC_DEFLATE "Build with DFLTCC intrinsics for compression on IBM Z" OFF)
     option(WITH_DFLTCC_INFLATE "Build with DFLTCC intrinsics for decompression on IBM Z" OFF)
     option(WITH_CRC32_VX "Build with vectorized CRC32 on IBM Z" ON)
+elseif(BASEARCH_X86_FOUND AND ${CMAKE_SYSTEM_PROCESSOR} STREQUAL "e2k")
+    option(WITH_AVX2 "Build with AVX2" OFF)
+    option(WITH_AVX512 "Build with AVX512" OFF)
+    option(WITH_AVX512VNNI "Build with AVX512 VNNI extensions" OFF)
+    option(WITH_SSE2 "Build with SSE2" ON)
+    option(WITH_SSSE3 "Build with SSSE3" ON)
+    option(WITH_SSE42 "Build with SSE42" OFF)
+    option(WITH_PCLMULQDQ "Build with PCLMULQDQ" ON)
+    option(WITH_VPCLMULQDQ "Build with VPCLMULQDQ" OFF)
+    if(NOT WITH_AVX2)
+        add_compile_options(-mno-avx)
+    endif()
 elseif(BASEARCH_X86_FOUND)
     option(WITH_SSE2 "Build with SSE2" ON)
     cmake_dependent_option(WITH_SSSE3 "Build with SSSE3" ON "WITH_SSE2" OFF)
@@ -933,7 +945,7 @@ if(WITH_OPTIM)
         endif()
         if(WITH_PCLMULQDQ)
             check_pclmulqdq_intrinsics()
-            if(HAVE_PCLMULQDQ_INTRIN AND WITH_SSE42)
+            if(HAVE_PCLMULQDQ_INTRIN AND (WITH_SSE42 OR ${CMAKE_SYSTEM_PROCESSOR} STREQUAL "e2k"))
                 add_definitions(-DX86_PCLMULQDQ_CRC)
                 set(PCLMULQDQ_SRCS ${ARCHDIR}/crc32_pclmulqdq.c)
                 add_feature_info(PCLMUL_CRC 1 "Support CRC hash generation using PCLMULQDQ, using \"${SSE42FLAG} ${PCLMULFLAG}\"")
diff --git a/arch/x86/x86_features.c b/arch/x86/x86_features.c
index 58cb4df..2e605cd 100644
--- a/arch/x86/x86_features.c
+++ b/arch/x86/x86_features.c
@@ -10,6 +10,19 @@
 #include "zbuild.h"
 #include "x86_features.h"
 
+#ifdef __e2k__
+void Z_INTERNAL x86_check_features(struct x86_cpu_features *features) {
+    features->has_sse2 = 1;
+    features->has_ssse3 = 1;
+#if __iset__ >= 6
+    features->has_pclmulqdq = 1;
+#endif
+#ifdef __AVX2__
+    features->has_os_save_ymm = 1;
+    features->has_avx2 = 1;
+#endif
+}
+#else
 #ifdef _MSC_VER
 #  include <intrin.h>
 #else
@@ -113,3 +126,4 @@ void Z_INTERNAL x86_check_features(struct x86_cpu_features *features) {
         }
     }
 }
+#endif
diff --git a/arch/x86/x86_intrins.h b/arch/x86/x86_intrins.h
index 0e596d1..a84783f 100644
--- a/arch/x86/x86_intrins.h
+++ b/arch/x86/x86_intrins.h
@@ -7,7 +7,12 @@
 #ifdef __AVX2__
 #include <immintrin.h>
 
-#if (!defined(__clang__) && !defined(__NVCOMPILER) && defined(__GNUC__) && __GNUC__ < 10) \
+#ifdef __e2k__
+#define _mm256_zextsi128_si256 _mm256_zextsi128_si256_redef
+static inline __m256i _mm256_zextsi128_si256(__m128i a) {
+    return _mm256_set_m128i(_mm_setzero_si128(), a);
+}
+#elif (!defined(__clang__) && !defined(__NVCOMPILER) && defined(__GNUC__) && __GNUC__ < 10) \
     || (defined(__apple_build_version__) && __apple_build_version__ < 9020039)
 static inline __m256i _mm256_zextsi128_si256(__m128i a) {
     __m128i r;
diff --git a/cmake/detect-arch.cmake b/cmake/detect-arch.cmake
index dfdc601..41a78cb 100644
--- a/cmake/detect-arch.cmake
+++ b/cmake/detect-arch.cmake
@@ -16,6 +16,8 @@ elseif(MSVC)
     elseif ("${MSVC_C_ARCHITECTURE_ID}" STREQUAL "ARM64" OR "${MSVC_C_ARCHITECTURE_ID}" STREQUAL "ARM64EC")
         set(ARCH "aarch64")
     endif()
+elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "e2k")
+    set(ARCH "x86_64")
 elseif(EMSCRIPTEN)
     set(ARCH "wasm32")
 elseif(CMAKE_CROSSCOMPILING)
diff --git a/cmake/detect-intrinsics.cmake b/cmake/detect-intrinsics.cmake
index 14f82fc..02e1bb7 100644
--- a/cmake/detect-intrinsics.cmake
+++ b/cmake/detect-intrinsics.cmake
@@ -274,6 +274,9 @@ macro(check_pclmulqdq_intrinsics)
         set(CMAKE_REQUIRED_FLAGS "${PCLMULFLAG} ${NATIVEFLAG} ${ZNOLTOFLAG}")
         check_c_source_compiles(
             "#include <immintrin.h>
+            #if defined(__e2k__) && __iset__ < 6
+            #error
+            #endif
             #include <wmmintrin.h>
             __m128i f(__m128i a, __m128i b) { return _mm_clmulepi64_si128(a, b, 0x10); }
             int main(void) { return 0; }"
-- 
2.34.1