Commit cb82feee authored by oscar's avatar oscar

提交编译

parent 3216c4e3
......@@ -79,9 +79,12 @@ add_definitions(-D_KF_IOU_CUDA_)
find_package(CUDA REQUIRED)
include_directories(/usr/local/cuda/include)
link_directories(/usr/local/cuda/lib64)
#set(CUDA_NVCC_PLAGS ${CUDA_NVCC_PLAGS};-std=c++11;-g;-G;-gencode;arch=compute_75;code=sm_75)
set(CUDA_NVCC_PLAGS ${CUDA_NVCC_PLAGS};-std=c++11;-g;-G;-gencode;arch=compute_75;code=sm_75)
include_directories(${PROJECT_SOURCE_DIR}/src/BaseTracker/kf_gpu)
cuda_add_library(kf_cuda SHARED ${PROJECT_SOURCE_DIR}/src/BaseTracker/kf_gpu/bev_overlap_online.cu ${PROJECT_SOURCE_DIR}/src/BaseTracker/kf_gpu/kalman_batch_ops.cu ${PROJECT_SOURCE_DIR}/src/BaseTracker/kf_gpu/kalman_update_batch_online.cu)
cuda_add_library(kf_cuda SHARED
src/BaseTracker/kf_gpu/bev_overlap_online.cu
src/BaseTracker/kf_gpu/kalman_batch_ops.cu
src/BaseTracker/kf_gpu/kalman_update_batch_online.cu)
target_link_libraries(kf_cuda cudart cublas)
endif()
......@@ -136,7 +139,7 @@ target_link_libraries(${PROJECT_NAME}_node
)
if(KF_CUDA)
target_link_libraries(${PROJECT_NAME}_node kf_cuda)
target_link_libraries(${PROJECT_NAME}_node kf_cuda cudart cublas)
endif()
#add_executable(cv_test
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment