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

silx-kit · Nov 9, 2022 · 023df28 · 023df28
1 parent bd177f8
commit 023df28
Showing 1 changed file with 9 additions and 1 deletion.
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
@@ -600,7 +609,6 @@ def prefix(directory, files):
     extra_link_args=extra_link_args,
     # TODO sse2=sse2_kwargs,
     # TODO avx2=avx2_kwargs,
-    # TODO cpp11=cpp11_kwargs,
     )
 
 # blosc plugin