Skip to content

Instantly share code, notes, and snippets.

@guschmue
Last active June 19, 2017 21:53
Show Gist options
  • Save guschmue/4b4178addcc96a4ecc7f03659b59bcd9 to your computer and use it in GitHub Desktop.
Save guschmue/4b4178addcc96a4ecc7f03659b59bcd9 to your computer and use it in GitHub Desktop.
cmake for user_ops
cmake_minimum_required(VERSION 3.5)
project(user_ops C CXX)
set(CMAKE_CXX_STANDARD 14)
set (tensorflow_SOURCE_DIR "c:/src/tensorflow.guschmue")
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
add_definitions(/I${tensorflow_SOURCE_DIR} /I${tensorflow_SOURCE_DIR}/cmake_build)
add_definitions(-DEIGEN_AVOID_STL_ARRAY)
add_definitions(-DNOMINMAX -D_WIN32_WINNT=0x0A00 -DLANG_CXX11 -DCOMPILER_MSVC)
add_definitions(-DWIN32 -DOS_WIN -D_MBCS -DWIN64 -DWIN32_LEAN_AND_MEAN -DNOGDI -DPLATFORM_WINDOWS)
add_definitions(-DTENSORFLOW_USE_EIGEN_THREADPOOL -DEIGEN_HAS_C99_MATH -D_ITERATOR_DEBUG_LEVEL=0)
add_definitions(/bigobj /nologo /EHsc /GF /FC /MP /Gm-)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /MP -DTF_EXTERN=\"extern __declspec(dllimport)\"")
include_directories(${tensorflow_SOURCE_DIR})
include_directories(${tensorflow_SOURCE_DIR}/cmake_build)
include_directories(${tensorflow_SOURCE_DIR}/cmake_build/protobuf/src/protobuf/src)
include_directories(${tensorflow_SOURCE_DIR}/cmake_build/external/eigen_archive)
include_directories(${tensorflow_SOURCE_DIR}/third_party/eigen3)
set (pywrap_tensorflow_lib "${tensorflow_SOURCE_DIR}/cmake_build/RelWithDebInfo/pywrap_tensorflow_internal.lib")
add_library(ackermann_op SHARED "ackermann_op.cc")
target_link_libraries(ackermann_op ${pywrap_tensorflow_lib})
add_custom_command(TARGET ackermann_op POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:ackermann_op> ackermann_op.so)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment