From a728be9cdb993b7a5bb9a842a6b1403d5d03462c Mon Sep 17 00:00:00 2001 From: Chidi Williams Date: Fri, 11 Nov 2022 16:10:01 +0000 Subject: [PATCH] Add WHISPER_NO_AVX and WHISPER_NO_AVX2 to CMakeLists (#136) * Check for AVX and AVX2 on Darwin * Add AVX options to CMakeLists --- CMakeLists.txt | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 38e4cd7..301143d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -48,6 +48,8 @@ option(WHISPER_SUPPORT_SDL2 "whisper: support for libSDL2" OFF) if (APPLE) option(WHISPER_NO_ACCELERATE "whisper: disable Accelerate framework" OFF) + option(WHISPER_NO_AVX "whisper: disable AVX" OFF) + option(WHISPER_NO_AVX2 "whisper: disable AVX2" OFF) else() option(WHISPER_SUPPORT_OPENBLAS "whisper: support for OpenBLAS" OFF) endif() @@ -151,7 +153,13 @@ else() set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -pthread -msimd128") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -pthread") else() - set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mavx -mavx2 -mfma -mf16c") + if(NOT WHISPER_NO_AVX) + set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mavx") + endif() + if(NOT WHISPER_NO_AVX2) + set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mavx2") + endif() + set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -mfma -mf16c") endif() endif() endif()