added AVX for linear basis function

CMakeLists.txt

@@ -2,22 +2,27 @@ cmake_minimum_required(VERSION 3.2)
 project(NeuralNetwork CXX)
 
 OPTION(BUILD_SHARED_LIBS "Build also shared library." ON)
+OPTION(USE_AVX "IF avx should be used." ON)
 OPTION(USE_SSE "IF sse should be used." ON)
 OPTION(USE_SSE2 "IF only sse2 should be used." OFF)
+
 OPTION(ENABLE_TESTS "enables tests" ON)
 
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -Wextra -pedantic -Weffc++ -Wshadow -Wstrict-aliasing -ansi -Woverloaded-virtual -Wdelete-non-virtual-dtor -Wno-unused-function")
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} --std=c++14")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g")
 #set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -pthread")
 
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=native -mtune=native -O3")
 
-if(USE_SSE)
+if(USE_AVX)
+    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DUSE_AVX")
+elseif(USE_SSE)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -msse4.2 -DUSE_SSE")
     if(USE_SSE2)
         set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DUSE_SSE2")
     endif(USE_SSE2)
-endif(USE_SSE)
+endif(USE_AVX)
 
 include_directories(./include/)