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

Seperate source files by their roles

parent b41fc37b
...@@ -5,7 +5,7 @@ string(REGEX REPLACE "^[0-9]+_" "" TRIMMED_DIR_NAME "${DIR_NAME}") ...@@ -5,7 +5,7 @@ string(REGEX REPLACE "^[0-9]+_" "" TRIMMED_DIR_NAME "${DIR_NAME}")
# add prefix "tile_example_" to the processed directory name # add prefix "tile_example_" to the processed directory name
set(EXAMPLE_NAME "tile_example_${TRIMMED_DIR_NAME}") set(EXAMPLE_NAME "tile_example_${TRIMMED_DIR_NAME}")
add_executable(${EXAMPLE_NAME} EXCLUDE_FROM_ALL paged_attention.cpp attention_launcher.cpp) add_executable(${EXAMPLE_NAME} EXCLUDE_FROM_ALL main.cpp itfs/paged_attention.cpp)
target_include_directories(${EXAMPLE_NAME} AFTER PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/include) target_include_directories(${EXAMPLE_NAME} AFTER PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/include)
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
......
#include "kernel/paged_attention_kernel.hpp"
#include "kernel/paged_attention_kernel.hpp"
#include <iostream> #include <iostream>
#include "paged_attention.hpp"
int main() { std::cout << "hello ck-tile example" << std::endl; } int main() { std::cout << "hello ck-tile example" << std::endl; }
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