Packages

torchx

0.2.1
0.13.0 0.12.0 0.11.0 0.10.2 0.10.1 0.10.0 0.9.2 0.9.1 0.9.0 0.8.0 0.7.3 0.7.2 0.7.1 0.7.0 0.6.4 0.6.3 0.6.1 0.6.0 0.5.3 0.5.2 0.5.1 0.5.0 0.4.2 0.4.1 0.4.0 0.3.0 0.2.1 0.2.0

LibTorch bindings and backend for Nx

HexDocs

Current section

Files

Jump to
Readme
28 Versions
3 Dependencies
Dependants
Files
Activity
torchx CMakeLists.txt
Raw

CMakeLists.txt 

cmake_minimum_required(VERSION 3.12 FATAL_ERROR)

project(torchx)



find_package(Torch REQUIRED)



STRING(REGEX REPLACE "\\\\" "/" C_SRC ${C_SRC})



file(GLOB torchx_sources CONFIGURE_DEPENDS "${C_SRC}/*.cpp" "${C_SRC}/*.hpp")



if (NOT APPLE)

add_library(torchx SHARED ${torchx_sources})

else()

add_library(torchx MODULE ${torchx_sources})

endif()



target_link_libraries(torchx "${TORCH_LIBRARIES}")

set_property(TARGET torchx PROPERTY CXX_STANDARD 14)



target_include_directories(torchx PUBLIC ${ERTS_INCLUDE_DIR})



SET_TARGET_PROPERTIES(torchx PROPERTIES PREFIX "")



if(UNIX)

set_target_properties(torchx PROPERTIES

    INSTALL_RPATH_USE_LINK_PATH TRUE

    BUILD_WITH_INSTALL_RPATH TRUE

)



if(NOT APPLE)

    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -shared")

    set_target_properties(torchx PROPERTIES INSTALL_RPATH "\$ORIGIN/${LIBTORCH_BASE}")

else()

    # Although the compiler complains about not using these,

    # things only work with them set

    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -undefined dynamic_lookup")

    # set(CMAKE_SHARED_LINKER_FLAGS "-bundle -flat_namespace -undefined suppress")

    set_target_properties(torchx PROPERTIES INSTALL_RPATH "@loader_path/${LIBTORCH_BASE}")

endif()



set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -O3 -Wall -Wextra -Wno-unused-parameter -Wno-missing-field-initializers")

else()

# On Windows

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /wd4624")

endif()