CMakeLists.txt

CMAKE_MINIMUM_REQUIRED(VERSION 2.8 FATAL_ERROR)
CMAKE_POLICY(VERSION 2.8)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D_FORCE_INLINES")
SET(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake ${CMAKE_MODULE_PATH})

FIND_PACKAGE(Torch REQUIRED)
FIND_PACKAGE(CUDA 5.5 REQUIRED)
FIND_PACKAGE(MAGMA)

IF (NOT WIN32)
SET(CMAKE_C_FLAGS "-std=c99 -Werror=implicit-function-declaration")
ENDIF (NOT WIN32)

INCLUDE_DIRECTORIES(${CUDA_INCLUDE_DIRS})

ADD_SUBDIRECTORY(lib)

INCLUDE_DIRECTORIES(BEFORE "${CMAKE_CURRENT_BINARY_DIR}/lib/THC")
INCLUDE_DIRECTORIES("${CMAKE_CURRENT_SOURCE_DIR}/lib/THC")
INCLUDE_DIRECTORIES("${CMAKE_CURRENT_SOURCE_DIR}/torch")
INCLUDE_DIRECTORIES("/home/philipp/torch/install/include/THC")
INCLUDE_DIRECTORIES("${THC_INSTALL_INCLUDE_SUBDIR}/THC")

SET(src Storage.c init.c Tensor.c TensorMath.c TensorOperator.c torch/utils.c)
SET(luasrc init.lua Tensor.lua FFI.lua test/test.lua THZCi.lua)

ADD_TORCH_WRAP(zcudatensormathwrap TensorMath.lua)

ADD_TORCH_PACKAGE(zcutorch "${src}" "${luasrc}")
TARGET_LINK_LIBRARIES(zcutorch luaT THZ THC THZC)

IF(LUALIB)
  TARGET_LINK_LIBRARIES(zcutorch ${LUALIB})
ENDIF()