Skip to content

GitLab

Commit 34fe00cb authored by Po Yen, Chen's avatar Po Yen, Chen
Browse files

Link torch libraires in example

parent ab26a544
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
example/ck_tile/18_paged_attention/CMakeLists.txt
View file @ 34fe00cb
...@@ -42,11 +42,15 @@ endif() ...@@ -42,11 +42,15 @@ endif()
# Use find_package() to locate Torch # Use find_package() to locate Torch
find_package(Torch REQUIRED) find_package(Torch REQUIRED)
add_executable(${EXAMPLE_NAME} EXCLUDE_FROM_ALL main.cpp itfs/paged_attention.cpp) add_executable(${EXAMPLE_NAME} EXCLUDE_FROM_ALL
main.cpp
itfs/paged_attention.cpp
)
target_include_directories(${EXAMPLE_NAME} target_include_directories(${EXAMPLE_NAME}
AFTER PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/include) AFTER PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/include)
target_include_directories(${EXAMPLE_NAME} target_include_directories(${EXAMPLE_NAME}
SYSTEM PRIVATE ${TORCH_INCLUDE_DIRS}) SYSTEM PRIVATE ${TORCH_INCLUDE_DIRS})
target_link_libraries(${EXAMPLE_NAME} "${TORCH_LIBRARIES}")
target_compile_definitions(${EXAMPLE_NAME} PRIVATE USE_ROCM) target_compile_definitions(${EXAMPLE_NAME} PRIVATE USE_ROCM)
target_compile_options(${EXAMPLE_NAME} PRIVATE target_compile_options(${EXAMPLE_NAME} PRIVATE
${TORCH_CXX_FLAGS} ${TORCH_CXX_FLAGS}
...@@ -56,4 +60,5 @@ target_compile_options(${EXAMPLE_NAME} PRIVATE ...@@ -56,4 +60,5 @@ target_compile_options(${EXAMPLE_NAME} PRIVATE
-Wno-old-style-cast -Wno-old-style-cast
-Wno-deprecated-copy -Wno-deprecated-copy
-Wno-shadow -Wno-shadow
-Wno-conditional-uninitialized
) )
\ No newline at end of file
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