mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 05:31:24 +00:00
20306db651
* CK tile tests for flatmm using example * MOE smoothquant draft tests * fix create_arg default index to zero for MOE smoothquant * revert MOE smoothquant changes * code clean up * Add back MOE smoothquant changes * Add MOE smoothquant cases for different precisions and update cmake * clean up comments * Update flamm cmake * revert change made to moe_smoothquant smoke_test.sh EXE path * remove unecessary comment in MOE smoothquant cmakelist * comment out adding moe_smoothquant subdirectory for now due to bugs with GPU core dump issue on gfx942 and gfx90a * Clean up run_test_case function in MOE smootquant tests * update copyright and licensing on files * Remove flatmm test dir since tests should be done as weighted preshuffle gemm * Add flamm smoke test cases to weighted preshuffle gemm gtests * remove blank line from CMakeLists --------- Co-authored-by: root <root@ctr-ubbsmc16.amd.com> Co-authored-by: Thomas Ning <Thomas.Ning@amd.com>
32 lines
1.5 KiB
CMake
32 lines
1.5 KiB
CMake
# Currently ck_tile is only built on gfx9
|
|
if(GPU_TARGETS MATCHES "gfx9")
|
|
function (add_moe_smoothquant_test TARGET_NAME MAIN_SRC)
|
|
message(DEBUG "adding ${TARGET_NAME}")
|
|
add_test_executable(${TARGET_NAME} ${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)
|
|
# list(APPEND COMPILE_OPTIONS -v --save-temps -Wno-gnu-line-marker)
|
|
|
|
target_compile_options(${TARGET_NAME} PRIVATE ${COMPILE_OPTIONS})
|
|
endfunction(add_moe_smoothquant_test TARGET_NAME MAIN_SRC)
|
|
|
|
file(GLOB INSTANCE_SRCS instances/*.cpp)
|
|
|
|
add_moe_smoothquant_test(test_ck_tile_moe_smoothquant_fp16_fp8 moe_smoothquant_fp16_fp8.cpp ${INSTANCE_SRCS})
|
|
add_moe_smoothquant_test(test_ck_tile_moe_smoothquant_fp16_int8 moe_smoothquant_fp16_int8.cpp ${INSTANCE_SRCS})
|
|
|
|
add_moe_smoothquant_test(test_ck_tile_moe_smoothquant_bf16_fp8 moe_smoothquant_bf16_fp8.cpp ${INSTANCE_SRCS})
|
|
add_moe_smoothquant_test(test_ck_tile_moe_smoothquant_bf16_int8 moe_smoothquant_bf16_int8.cpp ${INSTANCE_SRCS})
|
|
|
|
else()
|
|
message(DEBUG "Skipping ck_tile MOE smoothquant tests for current target")
|
|
endif() |