From ec40b70ea91dd59ba41817347524dd46df5b5512 Mon Sep 17 00:00:00 2001 From: netrunnereve <139727413+netrunnereve@users.noreply.github.com> Date: Thu, 2 Nov 2023 22:06:32 -0400 Subject: [PATCH] linux/gcc version for testing --- cmake/FindSIMD.cmake | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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()