ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 13:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK-tile] add gtest for ck-tile batched transpose kernels (#2585)

Browse Source 
* add a dummy test file

* add kernel launch logic to the test

* transfer all test cases into gtest params

* factor kernel out into test config

* add load transpose pipeline tests

* add padded tests and skip invalid kernels at runtime

* enum class for pipeline type

* add multiwarp test cases

* fix type

* try to solve the problem

---------

Co-authored-by: ThomasNing <thomas.ning@amd.com>
This commit is contained in:
Max Podkorytov
2025-07-30 07:31:05 -07:00
committed by GitHub
parent b25d512e8a
commit de0cdb4c31
2 changed files with 265 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/ck_tile/batched_transpose/CMakeLists.txt
View File

@@ -26,8 +26,8 @@ if(GPU_TARGETS MATCHES "gfx9")
add_batched_transpose_test(test_ck_tile_batched_transpose_bf16 batched_transpose_bf16.cpp)
add_dependencies(${CUSTOM_TARGET_NAME} test_ck_tile_batched_transpose_bf16)
add_gtest_executable(test_batched_transpose test_batched_transpose.cpp)
set_property(TARGET test_batched_transpose PROPERTY CXX_STANDARD 20)
else()
message(DEBUG "Skipping ck_tile batched_transpose tests for current target")
endif()