mirror of https://github.com/llvm/torch-mlir
23 lines
530 B
CMake
23 lines
530 B
CMake
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}
|
|
${PYTHON_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}
|
|
${PYTHON_LIBRARIES}
|
|
torch_python
|
|
)
|