mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-06-01 11:52:14 +00:00
12 lines
408 B
CMake
12 lines
408 B
CMake
project(tgiccl LANGUAGES C CXX CUDA)
|
|
|
|
set(TGICCL_HEADERS tgiccl.hpp TgiCclBackend.hpp)
|
|
set(TGICCL_SOURCES TgiCclBackend.cpp)
|
|
|
|
find_package(CUDAToolkit REQUIRED)
|
|
|
|
add_library(tgiccl SHARED ${TGICCL_HEADERS} ${TGICCL_SOURCES})
|
|
target_link_libraries(tgiccl PUBLIC spdlog::spdlog CUDA::nvml ${TORCH_LIBRARIES})
|
|
|
|
add_executable(test_tgiccl test_tgiccl.cpp)
|
|
target_link_libraries(test_tgiccl tgiccl spdlog::spdlog) |