Add multi-threads support to SAFT_TOFI

This commit is contained in:
kradchen
2023-06-19 17:06:16 +08:00
parent 06d293fc58
commit c4c22ff6cf
2 changed files with 19 additions and 11 deletions

View File

@@ -2,7 +2,7 @@ cmake_minimum_required(VERSION 3.8 FATAL_ERROR)
project(SaftTofi)
set(CMAKE_CUDA_COMPILER /usr/local/cuda/bin/nvcc)
enable_language(CUDA)
find_package (OpenMP REQUIRED)
add_library(SaftTofi SHARED ./src/SAFT_TOFI.cpp ./src/saft.cu ./src/processAScans.cpp ./src/saft.cpp )
target_include_directories(SaftTofi PRIVATE ../SAFT ./src /usr/local/cuda/include )
set_target_properties(SaftTofi PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
@@ -14,6 +14,8 @@ target_compile_options(SaftTofi PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:
>)
target_link_libraries(SaftTofi PRIVATE ${CUDA_RUNTIME_LIBRARY} )
target_link_libraries(SaftTofi PRIVATE OpenMP::OpenMP_CXX )
get_filename_component(_DIR "${CMAKE_CURRENT_LIST_DIR}" PATH)
set(SAFT_HEADER ${CMAKE_CURRENT_LIST_DIR}/src/SAFT_TOFI.h ${_DIR}/SAFT/SAFTStructs.h)
set_target_properties(SaftTofi PROPERTIES PUBLIC_HEADER "${SAFT_HEADER}")