diff options
author | Sam Gross <sgross@fb.com> | 2016-01-05 07:57:45 +0300 |
---|---|---|
committer | Sam Gross <sgross@fb.com> | 2016-01-05 22:45:11 +0300 |
commit | a142233b8bebc7a4acfa0ad6a66c1d400803034e (patch) | |
tree | bcef518cdd9125c5c68d493bbfc5873261b72192 /CMakeLists.txt | |
parent | 31f71b92881ad1ef3356a5077725f0541f765340 (diff) |
Add C implementation of SpatialBatchNormalization
This is primarily to support the fast, memory-efficient CUDA
implementation. Some other changes include making weight and bias each
individually optional and averaging the variances instead of the
inverse standard deviation.
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 535cdc0..69e18e8 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -10,6 +10,12 @@ IF(MSVC) ADD_DEFINITIONS(-D_CRT_SECURE_NO_DEPRECATE=1) ENDIF(MSVC) +IF (CMAKE_VERSION VERSION_LESS "3.1") + SET(CMAKE_C_FLAGS "--std=c99 ${CMAKE_C_FLAGS}") +ELSE () + SET(CMAKE_C_STANDARD 99) +ENDIF () + # OpenMP support? SET(WITH_OPENMP ON CACHE BOOL "OpenMP support if available?") IF (APPLE AND CMAKE_COMPILER_IS_GNUCC) |