Adapt to rocm 不适用flashattention2

6df4a6ac · xiabo · bc3c64aa · 6df4a6ac
Commit 6df4a6ac authored Nov 15, 2023 by xiabo
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

examples/cpp/llama/CMakeLists.txt examples/cpp/llama/CMakeLists.txt +2 -1

No files found.
--- a/examples/cpp/llama/CMakeLists.txt
+++ b/examples/cpp/llama/CMakeLists.txt
 # 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
+#target_link_libraries(llama_triton_example PUBLIC -lcublas -lcublasLt -lcudart
+target_link_libraries(llama_triton_example PUBLIC -lcublas -lcudart
        LlamaTritonBackend TransformerTritonBackend mpi_utils nccl_utils
        nvtx_utils word_list -lpthread)