CMakeLists.txt 381 Bytes
Newer Older
Li Zhang's avatar
Li Zhang committed
1
2
3
4
# Copyright (c) OpenMMLab. All rights reserved.

add_executable(llama_triton_example llama_triton_example.cc)
target_link_libraries(llama_triton_example PUBLIC -lcublas -lcublasLt -lcudart
AllentDan's avatar
AllentDan committed
5
        LlamaTritonBackend TransformerTritonBackend mpi_utils nccl_utils
Chen Xin's avatar
Chen Xin committed
6
        nvtx_utils word_list -lpthread)
7
8

install(TARGETS llama_triton_example DESTINATION ${CMAKE_INSTALL_PREFIX}/bin)