aes: initial implementation of most aes instructions (#1072)

simd-everywhere · Oct 13, 2023 · 8632391 · 8632391
1 parent aae2245
commit 8632391
Show file tree

Hide file tree

Showing 7 changed files with 983 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -106,6 +106,7 @@ jobs:
         - -mavx
         - -mfma
         - -mavx2
+        - -maes
         include:
           - target: tgl
           - isax: -mavx512bw -mavx512vl -mavx512cd -mavx512dq -mavx512vbmi -mavx512ifma -mavx512vpopcntdq -mavx512bitalg -mavx512vnni -mvpclmulqdq -mavx512fp16

diff --git a/meson.build b/meson.build
@@ -393,6 +393,7 @@ if not meson.is_subproject()
 
   install_headers(
     files([
+      'simde/x86/aes.h',
       'simde/x86/avx512.h',
       'simde/x86/avx2.h',
       'simde/x86/avx.h',

diff --git a/simde/simde-arch.h b/simde/simde-arch.h
@@ -339,6 +339,9 @@
 #  if defined(__F16C__) || (defined(HEDLEY_MSVC_VERSION) && HEDLEY_MSVC_VERSION_CHECK(19,30,0) && defined(SIMDE_ARCH_X86_AVX2) )
 #    define SIMDE_ARCH_X86_F16C 1
 #  endif
+#  if defined(__AES__)
+#    define SIMDE_ARCH_X86_AES 1
+#  endif
 #endif
 
 /* Itanium

diff --git a/simde/simde-features.h b/simde/simde-features.h
@@ -241,6 +241,15 @@
   #define SIMDE_X86_SSE2_NATIVE
 #endif
 
+#if !defined(SIMDE_X86_AES_NATIVE) && !defined(SIMDE_X86_AES_NO_NATIVE) && !defined(SIMDE_NO_NATIVE)
+  #if defined(SIMDE_ARCH_X86_AES)
+    #define SIMDE_X86_AES_NATIVE
+  #endif
+#endif
+#if defined(SIMDE_X86_AES_NATIVE) && !defined(SIMDE_X86_SSE2_NATIVE)
+  #define SIMDE_X86_SSE2_NATIVE
+#endif
+
 #if !defined(SIMDE_X86_SSE2_NATIVE) && !defined(SIMDE_X86_SSE2_NO_NATIVE) && !defined(SIMDE_NO_NATIVE)
   #if defined(SIMDE_ARCH_X86_SSE2)
     #define SIMDE_X86_SSE2_NATIVE
@@ -324,6 +333,10 @@
   #endif
 #endif
 
+#if defined(SIMDE_X86_AES_NATIVE)
+  #include <wmmintrin.h>
+#endif
+
 #if defined(HEDLEY_MSVC_VERSION)
   #pragma warning(pop)
 #endif
@@ -647,6 +660,9 @@
   #if !defined(SIMDE_X86_F16C_NATIVE)
     #define SIMDE_X86_F16C_ENABLE_NATIVE_ALIASES
   #endif
+  #if !defined(SIMDE_X86_AES_NATIVE)
+    #define SIMDE_X86_AES_ENABLE_NATIVE_ALIASES
+  #endif
 
   #if !defined(SIMDE_ARM_NEON_A32V7_NATIVE)
     #define SIMDE_ARM_NEON_A32V7_ENABLE_NATIVE_ALIASES

diff --git a/simde/x86/aes.h b/simde/x86/aes.h
diff --git a/test/x86/aes.c b/test/x86/aes.c
diff --git a/test/x86/meson.build b/test/x86/meson.build
@@ -13,7 +13,8 @@ simde_test_x86_tests = [
   'gfni',
   'clmul',
   'svml',
-  'xop'
+  'xop',
+  'aes'
 ]
 
 subdir('avx512')