diff --git a/cmake/FindSIMD.cmake b/cmake/FindSIMD.cmake index a7dd48890..596534cb9 100644 --- a/cmake/FindSIMD.cmake +++ b/cmake/FindSIMD.cmake @@ -77,21 +77,21 @@ macro(check_sse type flags) endmacro() -check_sse("AVX" " ;/arch:AVX") +check_sse("AVX" "-mavx") if(NOT ${AVX_FOUND}) set(LLAMA_AVX OFF) else() set(LLAMA_AVX ON) endif() -check_sse("AVX2" " ;/arch:AVX2") +check_sse("AVX2" "-mavx2") if(NOT ${AVX2_FOUND}) set(LLAMA_AVX2 OFF) else() set(LLAMA_AVX2 ON) endif() -check_sse("AVX512" " ;/arch:AVX512") +check_sse("AVX512" "-mavx512f -mavx512cd -mavx512bw -mavx512dq -mavx512vl") if(NOT ${AVX512_FOUND}) set(LLAMA_AVX512 OFF) else()