Skip to content

Commit 8cbd09c

Browse files
authored
[CK_TILE] Stream-K Tile Engine Test Config File Generation (#3662)
* Stream-K smoke test config file generation This change converts the stream-k smoke tests to use tile engine. Since the m, n, and k values dependent on the CU count of a device, the configs are generated during the Configuration Phase. * Compute GEMM reference on GPU * Remove redundant Stream-K tests Removing redundant tests that are now run via tile engine. * Fix relative and absolute tolerance calculation This change updates the Stream-K tile engine interface to ensure that num_wgs_per_tile is propaged and passed into the compare_results function to calculate the rel and abs tolerance. Before, split-k was used, which is incorrect for Stream-K since the split-k value is always 1. * Cleanup imports, types, and other misc items This commit makes the following changes: - Uses Typing module for nested type hints - Uses quotes around cu_count_arg argument in generate_configs.cmake in if statements - Adds explicit include for tuple in test_gemm_streamk_simple.cpp - Adds a type for the tiles argument in argparser to check argument validity * Use CU count as return value for better parsing * Add reduction tests for bf16, fp8, and bf8
1 parent 3f04d27 commit 8cbd09c

22 files changed

Lines changed: 523 additions & 407 deletions

test/ck_tile/gemm_streamk/CMakeLists.txt

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -23,19 +23,6 @@ if(GPU_TARGETS MATCHES "gfx90a|gfx942|gfx950")
2323
#TODO: support all arches
2424
#TODO: current c-shuffle only supports C layout as R
2525
add_gtest_executable(test_ck_tile_streamk_tile_partitioner test_streamk_tile_partitioner.cpp)
26-
add_gtest_executable(test_ck_tile_streamk_reduction
27-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_fp16_reduction.cpp
28-
test_gemm_streamk_util.cpp)
29-
add_gtest_executable(test_ck_tile_streamk_smoke
30-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_fp16_persistent.cpp
31-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_bf16_persistent.cpp
32-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_fp8_persistent.cpp
33-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_bf8_persistent.cpp
34-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_fp16_nonpersistent.cpp
35-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_bf16_nonpersistent.cpp
36-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_fp8_nonpersistent.cpp
37-
${CMAKE_CURRENT_SOURCE_DIR}/smoke_tests/test_gemm_streamk_bf8_nonpersistent.cpp
38-
test_gemm_streamk_util.cpp)
3926
add_gtest_executable(test_ck_tile_streamk_extended
4027
${CMAKE_CURRENT_SOURCE_DIR}/extended_tests/test_gemm_streamk_fp16_persistent.cpp
4128
${CMAKE_CURRENT_SOURCE_DIR}/extended_tests/test_gemm_streamk_bf16_persistent.cpp
@@ -46,7 +33,6 @@ if(GPU_TARGETS MATCHES "gfx90a|gfx942|gfx950")
4633
${CMAKE_CURRENT_SOURCE_DIR}/extended_tests/test_gemm_streamk_fp8_nonpersistent.cpp
4734
${CMAKE_CURRENT_SOURCE_DIR}/extended_tests/test_gemm_streamk_bf8_nonpersistent.cpp
4835
test_gemm_streamk_util.cpp)
49-
target_compile_options(test_ck_tile_streamk_smoke PRIVATE ${EXAMPLE_GEMM_COMPILE_OPTIONS})
5036
target_compile_options(test_ck_tile_streamk_extended PRIVATE ${EXAMPLE_GEMM_COMPILE_OPTIONS})
5137
else()
5238
message(DEBUG "Skipping test_ck_tile_streamk unit tests for current target")

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_bf16_nonpersistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_bf16_persistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_bf8_nonpersistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_bf8_persistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_fp16_nonpersistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_fp16_persistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_fp16_reduction.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_fp8_nonpersistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

test/ck_tile/gemm_streamk/smoke_tests/test_gemm_streamk_fp8_persistent.cpp

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)