Unverified Commit 8b95d9ad authored by Illia Silin's avatar Illia Silin Committed by GitHub
Browse files

copy all fmha headers when building library (#1497)

* copy all fmha headers when building library

* fix the rocm_install call for mha headers
parent 841009c5
...@@ -18,17 +18,10 @@ endif() ...@@ -18,17 +18,10 @@ endif()
rocm_install(DIRECTORY ${CK_TILE_SRC_FOLDER} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/ck_tile) rocm_install(DIRECTORY ${CK_TILE_SRC_FOLDER} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/ck_tile)
rocm_install(FILES file(GLOB MHA_HEADERS "${FMHA_SRC_FOLDER}/*.hpp")
"${FMHA_SRC_FOLDER}/fmha_fwd.hpp" rocm_install(FILES ${MHA_HEADERS} DESTINATION include/ck_tile/ops)
"${FMHA_SRC_FOLDER}/bias.hpp" # headers for building lib
"${FMHA_SRC_FOLDER}/mask.hpp" file(COPY ${MHA_HEADERS} DESTINATION ${FMHA_CPP_FOLDER})
DESTINATION include/ck_tile/ops
)
# header for building lib
file(COPY ${FMHA_SRC_FOLDER}/fmha_fwd.hpp DESTINATION ${FMHA_CPP_FOLDER})
file(COPY ${FMHA_SRC_FOLDER}/bias.hpp DESTINATION ${FMHA_CPP_FOLDER})
file(COPY ${FMHA_SRC_FOLDER}/mask.hpp DESTINATION ${FMHA_CPP_FOLDER})
# generate a list of kernels, but not actually emit files at config stage # generate a list of kernels, but not actually emit files at config stage
execute_process( execute_process(
......
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