Add see2/avx2 for blosc2 as for blosc(1)

silx-kit · Nov 9, 2022 · a9559f4 · a9559f4
1 parent bd177f8
commit a9559f4
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -553,6 +553,15 @@ def prefix(directory, files):
 extra_objects = []
 
 # TODO enable sse2/avx2/altivec/neon
+sse2_kwargs = {
+    'sources': [f for f in glob(blosc2_dir + 'blosc/*.c') if 'sse2' in f],
+    'define_macros': [('SHUFFLE_SSE2_ENABLED', 1)],
+    }
+
+avx2_kwargs = {
+    'sources': [f for f in glob(blosc2_dir + 'blosc/*.c') if 'avx2' in f],
+    'define_macros': [('SHUFFLE_AVX2_ENABLED', 1)],
+    }
 
 # compression libs
 # lz4
@@ -598,9 +607,8 @@ def prefix(directory, files):
     define_macros=define_macros,
     extra_compile_args=extra_compile_args,
     extra_link_args=extra_link_args,
-    # TODO sse2=sse2_kwargs,
-    # TODO avx2=avx2_kwargs,
-    # TODO cpp11=cpp11_kwargs,
+    sse2=sse2_kwargs,
+    avx2=avx2_kwargs,
     )
 
 # blosc plugin