include_directories(
  ${TORCH_INCLUDE_DIRS}
  ${TORCH_INSTALL_PREFIX}/include/TH
  ${TORCH_INSTALL_PREFIX}/include/THC/opt/pytorch/pytorch
  ${CMAKE_CURRENT_SOURCE_DIR}
  ${CMAKE_CURRENT_BINARY_DIR}
  ${Python3_INCLUDE_DIRS}
  )
link_directories("${TORCH_INSTALL_PREFIX}/lib")
add_library(npcomp_torch_builder_bindings
  acap_dispatch.cpp
  debug.cpp
  func_builder.cpp
  module_builder.cpp
  python_bindings.cpp
)

target_link_libraries(npcomp_torch_builder_bindings
  ${TORCH_LIBRARIES}
  ${Python3_LIBRARIES}
  torch_python
  )