torch-mlir/frontends/pytorch/csrc/builder/CMakeLists.txt

23 lines
532 B
CMake
Raw Normal View History

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
)