shin/NeuralNetworkLib
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

modified cmake to set USE_FMA

Browse Source
This commit is contained in:
Shin
2016-03-14 22:54:42 +01:00
parent a48298b342
commit 52ba015950
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
CMakeLists.txt
View File

@@ -7,6 +7,9 @@ OPTION(BUILD_SHARED_LIBS "Build also shared library." ON)
if(CPU_AVX_AVAILABLE)
SET(USE_AVX ON CACHE BOOL "")
if(CPU_FMA_AVAILABLE)
SET(USE_FMA ON CACHE BOOL "")
endif(CPU_FMA_AVAILABLE)
endif(CPU_AVX_AVAILABLE)
if(CPU_SSE3_AVAILABLE)
@@ -19,6 +22,8 @@ if(CPU_SSE3_AVAILABLE)
endif(CPU_SSE3_AVAILABLE)
OPTION(USE_AVX "If AVX instruction set should be used." OFF)
OPTION(USE_FMA "If FMA instruction set should be used." OFF)
OPTION(USE_SSE "If SSE instruction set should be used." OFF)
OPTION(USE_SSE2 "If SSE 2 instruction set should be used." ON)
@@ -34,6 +39,10 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=native -mtune=native -O3")
if(USE_AVX)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mavx -DUSE_AVX")
if(USE_FMA)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -mfma -DUSE_FMA")
endif(USE_FMA)
endif(USE_AVX)
if(USE_SSE)