add altivec support

silx-kit · Dec 14, 2022 · 1f9a9e8 · 1f9a9e8
1 parent ad42296
commit 1f9a9e8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -789,7 +789,7 @@ def get_blosc2_plugin():
     include_dirs = [blosc2_dir, f'{blosc2_dir}/blosc', f'{blosc2_dir}/include']
     define_macros = []
 
-    # TODO enable altivec/neon
+    # TODO enable neon
     sse2_kwargs = {
         'sources': glob(f'{blosc2_dir}/blosc/*-sse2.c'),
         'define_macros': [('SHUFFLE_SSE2_ENABLED', 1)],
@@ -800,6 +800,11 @@ def get_blosc2_plugin():
         'define_macros': [('SHUFFLE_AVX2_ENABLED', 1)],
         }
 
+    if platform.machine() == "ppc64le":  # altivec
+        sources += glob(f'{blosc2_dir}/blosc/*-altivec.c')
+        define_macros += [('SHUFFLE_ALTIVEC_ENABLED', 1)]
+
+
     # compression libs
     # lz4
     include_dirs += get_lz4_clib('include_dirs')