mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 21:21:22 +00:00
f102eedfb3
[CK_TILE] Add new ck tile unit test * Add new ck tile unit test smoke-gemm-universal * Add new ck tile unit test smoke-gemm-basic * Add new ck tile unit test topk_softmax * Add new ck tile unit test add_rmsnorm2d_rdquant_fwd
20 lines
966 B
CMake
20 lines
966 B
CMake
function(add_tile_topk_softmax_test SUFFIX)
|
|
set(TEST_NAME "test_ck_tile_topk_softmax_${SUFFIX}")
|
|
add_test_executable(${TEST_NAME} test_topk_softmax_${SUFFIX}.cpp test_topk_softmax_api.cpp)
|
|
target_include_directories(${TEST_NAME} PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/)
|
|
|
|
set(TEST_TOPK_SOFTMAX_COMPILE_OPTIONS)
|
|
# NOTE: we turn off undefined-func-template to let source compile without explicit declare function specializations
|
|
list(APPEND TEST_TOPK_SOFTMAX_COMPILE_OPTIONS -Wno-undefined-func-template -Wno-float-equal)
|
|
# list(APPEND TEST_TOPK_SOFTMAX_COMPILE_OPTIONS -v --save-temps -Wno-gnu-line-marker)
|
|
target_compile_options(${TEST_NAME} PRIVATE ${TEST_TOPK_SOFTMAX_COMPILE_OPTIONS})
|
|
endfunction()
|
|
|
|
# Currently ck_tile is only built on gfx9
|
|
if(GPU_TARGETS MATCHES "gfx9")
|
|
add_tile_topk_softmax_test(fp16)
|
|
add_tile_topk_softmax_test(bf16)
|
|
else()
|
|
message(DEBUG "Skipping tile topk_softmax tests for current target")
|
|
endif()
|