dmlc--dgl
Zhi Lin
35 行
1.3 KiB
CMake
35 行
1.3 KiB
CMake
cmake_minimum_required(VERSION 3.5)
|
|
|
|
project(featgraph C CXX)
|
|
message(STATUS "Start configuring project ${PROJECT_NAME}")
|
|
|
|
# Find CUDA
|
|
include(../cmake/util/FindCUDA.cmake)
|
|
find_cuda(ON)
|
|
message(STATUS "${CUDA_INCLUDE_DIRS}")
|
|
|
|
add_custom_target(
|
|
featgraph_kernel
|
|
COMMAND python ${CMAKE_CURRENT_SOURCE_DIR}/pack_featgraph.py
|
|
COMMENT "Creating featgraph kernels..."
|
|
)
|
|
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -O2 -fPIC")
|
|
file(GLOB FEATGRAPH_SRC
|
|
src/featgraph.cc
|
|
src/tvm_runtime_pack.cc
|
|
)
|
|
add_library(featgraph_runtime SHARED ${FEATGRAPH_SRC})
|
|
target_include_directories(featgraph_runtime PRIVATE ${CUDA_INCLUDE_DIRS})
|
|
target_include_directories(featgraph_runtime PRIVATE "./include")
|
|
target_include_directories(featgraph_runtime PRIVATE "../third_party/tvm/include")
|
|
target_include_directories(featgraph_runtime PRIVATE "../third_party/tvm/3rdparty/dmlc-core/include")
|
|
target_include_directories(featgraph_runtime PRIVATE "../third_party/tvm/3rdparty/dlpack/include")
|
|
target_link_libraries(featgraph_runtime "dl" # dynamic linking
|
|
${CUDA_CUDART_LIBRARY}
|
|
${CUDA_CUDA_LIBRARY}
|
|
${CUDA_NVRTC_LIBRARY})
|
|
add_dependencies(featgraph_runtime featgraph_kernel)
|
|
|
|
install(TARGETS featgraph_runtime LIBRARY DESTINATION lib)
|