dmlc--dgl
Quan (Andy) Gan
9a7235faf2
* first commit * some thoughts * move around * more commit * more fixes * now it uses torch allocator * fix symbol export error * fix * fixes * test fix * add script * building separate library per version * fix for vs2019 * more fixes * fix on windows build * update jenkinsfile * auto copy built dlls for windows * lint and installation guide update * fix * specify conda environment * set environment for ci * fix * fix * fix * fix again * revert * fix cmake * fix * switch to using python interpreter path * remove scripts * debug * oops sorry * Update index.rst * Update index.rst * copies automatically, no need for this * do not print message if library not found * tiny fixes * debug on nightly * replace add_compile_definitions to make CMake 3.5 happy * fix linking to wrong lib for multiple pytorch envs * changed building strategy * fix nightly * fix windows * fix windows again * setup bugfix * address comments * change README
42 行
1.8 KiB
CMake
42 行
1.8 KiB
CMake
cmake_minimum_required(VERSION 3.5)
|
|
project(tensoradapter_pytorch C CXX)
|
|
|
|
# Find PyTorch cmake files and PyTorch versions with the python interpreter $PYTHON_INTERP
|
|
# (or "python" if empty)
|
|
file(TO_NATIVE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/find_cmake.py FIND_CMAKE_PY)
|
|
if(NOT PYTHON_INTERP)
|
|
set(PYTHON_INTERP python)
|
|
endif()
|
|
message(STATUS "Using Python interpreter: ${PYTHON_INTERP}")
|
|
execute_process(
|
|
COMMAND ${PYTHON_INTERP} ${FIND_CMAKE_PY}
|
|
OUTPUT_VARIABLE TORCH_PREFIX_VER
|
|
OUTPUT_STRIP_TRAILING_WHITESPACE)
|
|
message(STATUS "find_cmake.py output: ${TORCH_PREFIX_VER}")
|
|
list(GET TORCH_PREFIX_VER 0 TORCH_PREFIX)
|
|
list(GET TORCH_PREFIX_VER 1 TORCH_VER)
|
|
message(STATUS "Configuring for PyTorch ${TORCH_VER}")
|
|
|
|
set(Torch_DIR "${TORCH_PREFIX}/Torch")
|
|
message(STATUS "Setting directory to ${Torch_DIR}")
|
|
find_package(Torch REQUIRED)
|
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${TORCH_C_FLAGS}")
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
|
|
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -O0 -g3 -ggdb")
|
|
set(TORCH_TARGET_NAME "tensoradapter_pytorch_${TORCH_VER}")
|
|
file(GLOB TA_TORCH_SRC *.cpp)
|
|
add_library(${TORCH_TARGET_NAME} SHARED "${TA_TORCH_SRC}")
|
|
|
|
message(STATUS "tensoradapter found PyTorch includes: ${TORCH_INCLUDE_DIRS}")
|
|
message(STATUS "tensoradapter found PyTorch lib: ${TORCH_LIBRARIES}")
|
|
|
|
target_include_directories(
|
|
${TORCH_TARGET_NAME} PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/../include")
|
|
target_include_directories(
|
|
${TORCH_TARGET_NAME} PRIVATE "${CMAKE_CURRENT_SOURCE_DIR}/../../third_party/dlpack/include")
|
|
target_include_directories(
|
|
${TORCH_TARGET_NAME} PRIVATE "${TORCH_INCLUDE_DIRS}")
|
|
target_link_libraries(${TORCH_TARGET_NAME} PRIVATE "${TORCH_LIBRARIES}")
|
|
set_property(TARGET ${TORCH_TARGET_NAME} PROPERTY CXX_STANDARD 14)
|
|
message(STATUS "Configured target ${TORCH_TARGET_NAME}")
|