disable avx on windows by default

test=develop
local_add_cudnn_lstm
peizhilin 7 years ago
parent 449406434e
commit cc319f64cb

@ -57,6 +57,8 @@ int main()
return 0;
}" SSE3_FOUND)
# disable AVX by default on windows
if(NOT WIN32)
# Check AVX
set(CMAKE_REQUIRED_FLAGS ${AVX_FLAG})
set(AVX_FOUND_EXITCODE 1 CACHE STRING "Result from TRY_RUN" FORCE)
@ -70,8 +72,6 @@ int main()
return 0;
}" AVX_FOUND)
# disable AVX2 by default on windows
if(NOT WIN32)
# Check AVX 2
set(CMAKE_REQUIRED_FLAGS ${AVX2_FLAG})
set(AVX2_FOUND_EXITCODE 1 CACHE STRING "Result from TRY_RUN" FORCE)
@ -83,7 +83,6 @@ if(NOT WIN32)
__m256i result = _mm256_abs_epi32 (a);
return 0;
}" AVX2_FOUND)
endif(NOT WIN32)
# Check AVX512F
set(CMAKE_REQUIRED_FLAGS ${AVX512F_FLAG})
@ -97,6 +96,7 @@ int main()
__m512i result = _mm512_abs_epi32 (a);
return 0;
}" AVX512F_FOUND)
endif(NOT WIN32)
set(CMAKE_REQUIRED_FLAGS ${CMAKE_REQUIRED_FLAGS_RETAINED})
mark_as_advanced(MMX_FOUND SSE2_FOUND SSE3_FOUND AVX_FOUND AVX2_FOUND AVX512F_FOUND)

Loading…
Cancel
Save