Add AVX512 accelerated 1D/3D LUTS (#1932)

* Initial AVX512 support Signed-off-by: Mark Reid <[email protected]> * Lut1DOp add AVX512 implementation Signed-off-by: Mark Reid <[email protected]> * Lut3DOp add AVX512 implementation Signed-off-by: Mark Reid <[email protected]> * Don't use SIMD if only 1 pixel is requested Signed-off-by: Mark Reid <[email protected]> * Remove #if, f16c is always available with AVX512 Signed-off-by: Mark Reid <[email protected]> * Cast pointers to __m512 instead of __m256 Signed-off-by: Mark Reid <[email protected]> * Use size method from vector being tested Signed-off-by: Mark Reid <[email protected]> * Add to help message that f16c is only used with AVX/AVX2 Signed-off-by: Mark Reid <[email protected]> * Clarify test case by using uint8 maxValue Signed-off-by: Mark Reid <[email protected]> --------- Signed-off-by: Mark Reid <[email protected]>
AcademySoftwareFoundation · Mar 21, 2024 · 91e8826 · 91e8826
1 parent 50e7d58
commit 91e8826
Show file tree

Hide file tree

Showing 17 changed files with 1,535 additions and 20 deletions.
diff --git a/share/cmake/utils/CheckSupportAVX512.cmake b/share/cmake/utils/CheckSupportAVX512.cmake
@@ -5,7 +5,7 @@ include(CheckCXXSourceCompiles)
 
 set(_cmake_cxx_flags_orig "${CMAKE_CXX_FLAGS}")
 
-if(APPLE AND ("${CMAKE_OSX_ARCHITECTURES}" MATCHES "arm64;x86_64" 
+if(APPLE AND ("${CMAKE_OSX_ARCHITECTURES}" MATCHES "arm64;x86_64"
           OR "${CMAKE_OSX_ARCHITECTURES}" MATCHES "x86_64;arm64"))
     set(__universal_build 1)
     set(_cmake_osx_architectures_orig "${CMAKE_OSX_ARCHITECTURES}")
@@ -28,6 +28,8 @@ set(AVX512_CODE "
 
     int main() {
         __m512i vec = _mm512_set1_epi32(42);
+        // gcc <= 6 appear to be missing this intrinsic
+        __mmask16 k = _mm512_int2mask(42);
         return 0;
     }
 ")

diff --git a/share/cmake/utils/CheckSupportX86SIMD.cmake b/share/cmake/utils/CheckSupportX86SIMD.cmake
@@ -31,6 +31,10 @@ if(MSVC)
     if (COMPILER_SUPPORTS_AVX2)
         set(OCIO_AVX2_ARGS "/arch:AVX2")
     endif()
+
+    if (COMPILER_SUPPORTS_AVX512)
+        set(OCIO_AVX512_ARGS "/arch:AVX512")
+    endif()     
 else()
     if (COMPILER_SUPPORTS_SSE2)
         set(OCIO_SSE2_ARGS "-msse2")
@@ -42,6 +46,10 @@ else()
 
     if (COMPILER_SUPPORTS_AVX2)
         set(OCIO_AVX2_ARGS "-mavx2" "-mfma")
+    endif()
+
+    if (COMPILER_SUPPORTS_AVX512)
+        set(OCIO_AVX512_ARGS "-mavx512f")
     endif()    
 endif()