@@ -42,9 +42,14 @@ IF(NOT WIN32)
4242 SET (MKLDNN_FLAG "${MKLDNN_FLAG} -Wno-unused-result -Wno-unused-value" )
4343 SET (MKLDNN_CFLAG "${CMAKE_C_FLAGS} ${MKLDNN_FLAG} " )
4444 SET (MKLDNN_CXXFLAG "${CMAKE_CXX_FLAGS} ${MKLDNN_FLAG} " )
45+ SET (MKLDNN_CXXFLAG_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} " )
46+ SET (MKLDNN_CFLAG_RELEASE "${CMAKE_C_FLAGS_RELEASE} " )
4547 SET (MKLDNN_LIB "${MKLDNN_INSTALL_DIR} /${LIBDIR} /libdnnl.so" CACHE FILEPATH "mkldnn library." FORCE)
4648ELSE ()
4749 SET (MKLDNN_CXXFLAG "${CMAKE_CXX_FLAGS} /EHsc" )
50+ SET (MKLDNN_CFLAG "${CMAKE_C_FLAGS} " )
51+ string (REPLACE "/O2 " "" MKLDNN_CFLAG_RELEASE "${CMAKE_C_FLAGS_RELEASE} " )
52+ string (REPLACE "/O2 " "" MKLDNN_CXXFLAG_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} " )
4853 SET (MKLDNN_LIB "${MKLDNN_INSTALL_DIR} /bin/mkldnn.lib" CACHE FILEPATH "mkldnn library." FORCE)
4954ENDIF (NOT WIN32 )
5055
@@ -60,17 +65,15 @@ ExternalProject_Add(
6065 #BUILD_ALWAYS 1
6166 CMAKE_ARGS -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER}
6267 -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER}
63- -DCMAKE_CXX_FLAGS_RELEASE=${CMAKE_CXX_FLAGS_RELEASE}
68+ -DCMAKE_CXX_FLAGS=${MKLDNN_CXXFLAG}
69+ -DCMAKE_CXX_FLAGS_RELEASE=${MKLDNN_CXXFLAG_RELEASE}
6470 -DCMAKE_CXX_FLAGS_DEBUG=${CMAKE_CXX_FLAGS_DEBUG}
65- -DCMAKE_C_FLAGS=${CMAKE_C_FLAGS }
71+ -DCMAKE_C_FLAGS=${MKLDNN_CFLAG }
6672 -DCMAKE_C_FLAGS_DEBUG=${CMAKE_C_FLAGS_DEBUG}
67- -DCMAKE_C_FLAGS_RELEASE=${CMAKE_C_FLAGS_RELEASE }
73+ -DCMAKE_C_FLAGS_RELEASE=${MKLDNN_CFLAG_RELEASE }
6874 -DCMAKE_INSTALL_PREFIX=${MKLDNN_INSTALL_DIR}
6975 -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE}
7076 -DCMAKE_POSITION_INDEPENDENT_CODE=ON
71- -DMKLROOT=${MKLML_ROOT}
72- -DCMAKE_C_FLAGS=${MKLDNN_CFLAG}
73- -DCMAKE_CXX_FLAGS=${MKLDNN_CXXFLAG}
7477 -DDNNL_BUILD_TESTS=OFF -DDNNL_BUILD_EXAMPLES=OFF
7578 CMAKE_CACHE_ARGS -DCMAKE_INSTALL_PREFIX:PATH =${MKLDNN_INSTALL_DIR}
7679)
0 commit comments