mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 05:01:25 +00:00
abae2afc72
* Fix cmake example build * Support max3 in smoothquant one pass * support max3 in two pass * support max3 in add_rmsnorm_rdquant
25 lines
1.1 KiB
CMake
25 lines
1.1 KiB
CMake
function (add_smoothquant_example TARGET_NAME MAIN_SRC)
|
|
message("adding ${TARGET_NAME}")
|
|
# not using add_example_executable() to add target, since we don't want this to have
|
|
# to be included in "make all/install/check"
|
|
add_executable(${TARGET_NAME} EXCLUDE_FROM_ALL ${MAIN_SRC})
|
|
target_include_directories(${TARGET_NAME} PRIVATE ${CMAKE_CURRENT_LIST_DIR})
|
|
|
|
foreach(source IN LISTS ARGN)
|
|
list(APPEND INSTANCE_SRCS ${source})
|
|
endforeach()
|
|
|
|
target_sources(${TARGET_NAME} PRIVATE ${INSTANCE_SRCS})
|
|
|
|
set(COMPILE_OPTIONS)
|
|
# NOTE: we turn off undefined-func-template to let source compile without explicit declare function specializations
|
|
list(APPEND COMPILE_OPTIONS -Wno-undefined-func-template -Wno-float-equal)
|
|
|
|
target_compile_options(${TARGET_NAME} PRIVATE ${COMPILE_OPTIONS})
|
|
endfunction(add_smoothquant_example TARGET_NAME MAIN_SRC)
|
|
|
|
|
|
add_smoothquant_example(tile_example_smoothquant example_smoothquant.cpp)
|
|
file(GLOB INSTANCE_SRCS instances/*.cpp)
|
|
add_smoothquant_example(tile_smoothquant smoothquant.cpp ${INSTANCE_SRCS})
|