mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-27 16:00:07 +08:00
erikwijmans
23 lines
755 B
CMake
23 lines
755 B
CMake
project(PointNet2)
|
|
cmake_minimum_required(VERSION 3.5)
|
|
|
|
find_package(CUDA REQUIRED)
|
|
|
|
include_directories("${CMAKE_SOURCE_DIR}/cinclude")
|
|
cuda_include_directories("${CMAKE_SOURCE_DIR}/cinclude")
|
|
file(GLOB cuda_kernels_src "csrc/*.cu")
|
|
cuda_compile(cuda_kernels SHARED ${cuda_kernels_src} OPTIONS -O3)
|
|
|
|
file(GLOB wrapper_headers "cinclude/*wrapper.h")
|
|
file(GLOB wrapper_sources "csrs/*.c")
|
|
add_custom_command(OUTPUT "${CMAKE_SOURCE_DIR}/_ext/__ext.so"
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}
|
|
COMMAND python "${CMAKE_SOURCE_DIR}/build_ffi.py" ${cuda_kernels}
|
|
DEPENDS ${cuda_kernels}
|
|
DEPENDS ${wrapper_headers}
|
|
DEPENDS ${wrapper_sources}
|
|
VERBATIM)
|
|
|
|
add_custom_target(ext ALL
|
|
DEPENDS "${CMAKE_SOURCE_DIR}/_ext/__ext.so")
|