mirror of
https://github.com/microsoft/vcpkg.git
synced 2024-11-24 14:29:06 +08:00
03037c40f7
Adding Torch's TH library
53 lines
2.8 KiB
Diff
53 lines
2.8 KiB
Diff
diff --git a/lib/TH/CMakeLists.txt b/lib/TH/CMakeLists.txt
|
|
index 803dfe1..7db876b 100644
|
|
--- a/lib/TH/CMakeLists.txt
|
|
+++ b/lib/TH/CMakeLists.txt
|
|
@@ -221,7 +221,13 @@ ENDIF(C_AVX2_FOUND OR C_AVX_FOUND OR C_SSE4_2_FOUND OR C_SSE4_1_FOUND)
|
|
IF(C_SSE4_1_FOUND AND C_SSE4_2_FOUND)
|
|
SET(CMAKE_C_FLAGS "${C_SSE4_1_FLAGS} -DUSE_SSE4_1 ${C_SSE4_2_FLAGS} -DUSE_SSE4_2 ${CMAKE_C_FLAGS}")
|
|
IF(MSVC)
|
|
- SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_sse.c PROPERTIES COMPILE_FLAGS "/Ox /fp:fast")
|
|
+ IF(CMAKE_BUILD_TYPE STREQUAL Release)
|
|
+ SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_sse.c PROPERTIES COMPILE_FLAGS "/Ox /fp:fast")
|
|
+ MESSAGE(STATUS "Release mode. Found SSE4_1 and SSE4_2.")
|
|
+ ELSE()
|
|
+ SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_sse.c PROPERTIES COMPILE_FLAGS "/fp:fast")
|
|
+ MESSAGE(STATUS "Non-release mode. Found SSE4_1 and SSE4_2.")
|
|
+ ENDIF()
|
|
ELSE(MSVC)
|
|
SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_sse.c PROPERTIES COMPILE_FLAGS "-O3 -ffast-math")
|
|
ENDIF(MSVC)
|
|
@@ -231,8 +237,15 @@ ENDIF(C_SSE4_1_FOUND AND C_SSE4_2_FOUND)
|
|
# IF AVX FOUND
|
|
IF(C_AVX_FOUND)
|
|
IF(MSVC)
|
|
- SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_avx.c PROPERTIES COMPILE_FLAGS "/Ox /fp:fast ${C_AVX_FLAGS}")
|
|
- SET_SOURCE_FILES_PROPERTIES(vector/AVX.c PROPERTIES COMPILE_FLAGS "/Ox /arch:AVX ${C_AVX_FLAGS}")
|
|
+ IF(CMAKE_BUILD_TYPE STREQUAL Release)
|
|
+ SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_avx.c PROPERTIES COMPILE_FLAGS "/Ox /fp:fast ${C_AVX_FLAGS}")
|
|
+ SET_SOURCE_FILES_PROPERTIES(vector/AVX.c PROPERTIES COMPILE_FLAGS "/Ox /arch:AVX ${C_AVX_FLAGS}")
|
|
+ MESSAGE(STATUS "Release mode. Found AVX.")
|
|
+ ELSE()
|
|
+ SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_avx.c PROPERTIES COMPILE_FLAGS "/fp:fast ${C_AVX_FLAGS}")
|
|
+ SET_SOURCE_FILES_PROPERTIES(vector/AVX.c PROPERTIES COMPILE_FLAGS "/arch:AVX ${C_AVX_FLAGS}")
|
|
+ MESSAGE(STATUS "Non-release mode. Found AVX.")
|
|
+ ENDIF()
|
|
ELSE(MSVC)
|
|
SET_SOURCE_FILES_PROPERTIES(generic/simd/convolve5x5_avx.c PROPERTIES COMPILE_FLAGS "-O3 -ffast-math ${C_AVX_FLAGS}")
|
|
SET_SOURCE_FILES_PROPERTIES(vector/AVX.c PROPERTIES COMPILE_FLAGS "-O3 ${C_AVX_FLAGS}")
|
|
@@ -242,7 +255,13 @@ ENDIF(C_AVX_FOUND)
|
|
|
|
IF(C_AVX2_FOUND)
|
|
IF(MSVC)
|
|
- SET_SOURCE_FILES_PROPERTIES(vector/AVX2.c PROPERTIES COMPILE_FLAGS "/Ox /arch:AVX2 ${C_AVX2_FLAGS}")
|
|
+ IF(CMAKE_BUILD_TYPE STREQUAL Release)
|
|
+ SET_SOURCE_FILES_PROPERTIES(vector/AVX2.c PROPERTIES COMPILE_FLAGS "/Ox /arch:AVX2 ${C_AVX2_FLAGS}")
|
|
+ MESSAGE(STATUS "Release mode. Found AVX2.")
|
|
+ ELSE()
|
|
+ SET_SOURCE_FILES_PROPERTIES(vector/AVX2.c PROPERTIES COMPILE_FLAGS "/arch:AVX2 ${C_AVX2_FLAGS}")
|
|
+ MESSAGE(STATUS "Non-release mode. Found AVX2.")
|
|
+ ENDIF()
|
|
ELSE(MSVC)
|
|
SET_SOURCE_FILES_PROPERTIES(vector/AVX2.c PROPERTIES COMPILE_FLAGS "-O3 ${C_AVX2_FLAGS}")
|
|
ENDIF(MSVC)
|