cmake: check nasm and define related MACRO

Signed-off-by: Changcheng Liu <changcheng.liu@aliyun.com>

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 8b515575896673587843e5dc987601a889f58585..ff64052c95c134ceeb7c86b91ae4601bcb410adb 100644 (file)
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -145,6 +145,16 @@ if(HAVE_INTEL)
     set(object_format "elf64")
   endif()
   set(CMAKE_ASM_FLAGS "-f ${object_format}")
+  set(CMAKE_ASM_COMPILER ${PROJECT_SOURCE_DIR}/src/nasm-wrapper)
+  if(NOT WIN32)
+    # The native tools might be located even when cross compiling, which
+    # might not work in this case (especially when targeting Windows).
+    include(CheckNasm)
+    check_nasm_support(${object_format}
+      HAVE_NASM_X64
+      HAVE_NASM_X64_AVX2
+      HAVE_NASM_X64_AVX512)
+  endif()
   set(CMAKE_ASM_COMPILER ${PROJECT_SOURCE_DIR}/src/yasm-wrapper)
   if(NOT WIN32)
     # The native tools might be located even when cross compiling, which

diff --git a/src/include/config-h.in.cmake b/src/include/config-h.in.cmake
index 65c1fba6a8689e2db877751b9abd9c435104843c..afc2449d80842eb15b2c8cb6b227fc581f9ae49a 100644 (file)
--- a/src/include/config-h.in.cmake
+++ b/src/include/config-h.in.cmake
@@ -250,6 +250,15 @@
 /* yasm can also build the isa-l:avx512 */
 #cmakedefine HAVE_YASM_X64_AVX512
 
+/* we have a recent nasm and are x86_64 */
+#cmakedefine HAVE_NASM_X64
+
+/* nasm can also build the isa-l:avx2 */
+#cmakedefine HAVE_NASM_X64_AVX2
+
+/* nasm can also build the isa-l:avx512 */
+#cmakedefine HAVE_NASM_X64_AVX512
+
 /* Define if isa-l is compiled for arm64 */
 #cmakedefine HAVE_ARMV8_SIMD