ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-18 01:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] Add supported grouped and batched GEMM test cases

Browse Source
This commit is contained in:
Aleksander Dudek
2025-09-16 10:48:58 +00:00
parent 59cb906482
commit 33ae2b65f7
2 changed files with 80 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
test/ck_tile/batched_gemm/test_batched_gemm.cpp
View File

@@ -8,8 +8,13 @@
#include "ck_tile/host.hpp"
#include "test_batched_gemm_util.hpp"
using F16 = ck_tile::half_t;
using F32 = float;
using INT8 = ck_tile::int8_t;
using INT32 = ck_tile::int32_t;
using F16 = ck_tile::half_t;
using F32 = float;
using F8 = ck_tile::fp8_t;
using BF8 = ck_tile::bf8_t;
using BF16 = ck_tile::bf16_t;
using Row = ck_tile::tensor_layout::gemm::RowMajor;
using Col = ck_tile::tensor_layout::gemm::ColumnMajor;
@@ -20,7 +25,23 @@ using KernelTypes = ::testing::Types<
std::tuple< Row, Row, Row, F16, F16, F32, F16>,
std::tuple< Col, Row, Row, F16, F16, F32, F16>,
std::tuple< Row, Col, Row, F16, F16, F32, F16>,
std::tuple< Col, Col, Row, F16, F16, F32, F16>
std::tuple< Col, Col, Row, F16, F16, F32, F16>,
std::tuple< Row, Row, Row, BF16, BF16, F32, F16>,
std::tuple< Col, Row, Row, BF16, BF16, F32, F16>,
std::tuple< Row, Col, Row, BF16, BF16, F32, F16>,
std::tuple< Col, Col, Row, BF16, BF16, F32, F16>,
std::tuple< Row, Row, Row, BF8, BF8, F32, F16>,
std::tuple< Col, Row, Row, BF8, BF8, F32, F16>,
std::tuple< Row, Col, Row, BF8, BF8, F32, F16>,
std::tuple< Col, Col, Row, BF8, BF8, F32, F16>,
std::tuple< Row, Row, Row, F8, F8, F32, F16>,
std::tuple< Col, Row, Row, F8, F8, F32, F16>,
std::tuple< Row, Col, Row, F8, F8, F32, F16>,
std::tuple< Col, Col, Row, F8, F8, F32, F16>,
std::tuple< Row, Row, Row, INT8, INT8, INT32, INT32>,
std::tuple< Col, Row, Row, INT8, INT8, INT32, INT32>,
std::tuple< Row, Col, Row, INT8, INT8, INT32, INT32>,
std::tuple< Col, Col, Row, INT8, INT8, INT32, INT32>
>;
// clang-format on

68
test/ck_tile/grouped_gemm/test_grouped_gemm.cpp
View File

@@ -8,8 +8,13 @@
#include "ck_tile/host.hpp"
#include "test_grouped_gemm_util.hpp"
using INT8 = ck_tile::int8_t;
using INT32 = ck_tile::int32_t;
using F16 = ck_tile::half_t;
using F32 = float;
using F8 = ck_tile::fp8_t;
using BF16 = ck_tile::bf16_t;
using BF8 = ck_tile::bf8_t;
using Row = ck_tile::tensor_layout::gemm::RowMajor;
using Col = ck_tile::tensor_layout::gemm::ColumnMajor;
using True = ck_tile::bool_constant<true>;
@@ -17,18 +22,57 @@ using False = ck_tile::bool_constant<false>;
// clang-format off
using KernelTypes = ::testing::Types<
// ALayout, BLayout, CLayout, ADataType, BDataType, AccDataType, CDataType, Persistent
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Col, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Col, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Row, Row, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Row, Row, F16, F16, F32, F16, False>,
std::tuple< Col, Row, Row, F16, F16, F32, F16, True>,
std::tuple< Col, Row, Row, F16, F16, F32, F16, False>
// ALayout, BLayout, CLayout, ADataType, BDataType, AccDataType, CDataType, Persistent
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Col, Col, Row, F16, F16, F32, F16, True>,
std::tuple< Col, Col, Row, F16, F16, F32, F16, False>,
std::tuple< Row, Row, Row, F16, F16, F32, F16, True>,
std::tuple< Row, Row, Row, F16, F16, F32, F16, False>,
std::tuple< Col, Row, Row, F16, F16, F32, F16, True>,
std::tuple< Col, Row, Row, F16, F16, F32, F16, False>,
std::tuple< Row, Col, Row, BF16, BF16, F32, F16, True>,
std::tuple< Row, Col, Row, BF16, BF16, F32, F16, False>,
std::tuple< Row, Col, Row, BF16, BF16, F32, F16, True>,
std::tuple< Row, Col, Row, BF16, BF16, F32, F16, False>,
std::tuple< Col, Col, Row, BF16, BF16, F32, F16, True>,
std::tuple< Col, Col, Row, BF16, BF16, F32, F16, False>,
std::tuple< Row, Row, Row, BF16, BF16, F32, F16, True>,
std::tuple< Row, Row, Row, BF16, BF16, F32, F16, False>,
std::tuple< Col, Row, Row, BF16, BF16, F32, F16, True>,
std::tuple< Col, Row, Row, BF16, BF16, F32, F16, False>,
std::tuple< Row, Col, Row, BF8, BF8, F32, F16, True>,
std::tuple< Row, Col, Row, BF8, BF8, F32, F16, False>,
std::tuple< Row, Col, Row, BF8, BF8, F32, F16, True>,
std::tuple< Row, Col, Row, BF8, BF8, F32, F16, False>,
std::tuple< Col, Col, Row, BF8, BF8, F32, F16, True>,
std::tuple< Col, Col, Row, BF8, BF8, F32, F16, False>,
std::tuple< Row, Row, Row, BF8, BF8, F32, F16, True>,
std::tuple< Row, Row, Row, BF8, BF8, F32, F16, False>,
std::tuple< Col, Row, Row, BF8, BF8, F32, F16, True>,
std::tuple< Col, Row, Row, BF8, BF8, F32, F16, False>,
std::tuple< Row, Col, Row, F8, F8, F32, F16, True>,
std::tuple< Row, Col, Row, F8, F8, F32, F16, False>,
std::tuple< Row, Col, Row, F8, F8, F32, F16, True>,
std::tuple< Row, Col, Row, F8, F8, F32, F16, False>,
std::tuple< Col, Col, Row, F8, F8, F32, F16, True>,
std::tuple< Col, Col, Row, F8, F8, F32, F16, False>,
std::tuple< Row, Row, Row, F8, F8, F32, F16, True>,
std::tuple< Row, Row, Row, F8, F8, F32, F16, False>,
std::tuple< Col, Row, Row, F8, F8, F32, F16, True>,
std::tuple< Col, Row, Row, F8, F8, F32, F16, False>,
std::tuple< Row, Col, Row, INT8, INT8, INT32, INT32, True>,
std::tuple< Row, Col, Row, INT8, INT8, INT32, INT32, False>,
std::tuple< Row, Col, Row, INT8, INT8, INT32, INT32, True>,
std::tuple< Row, Col, Row, INT8, INT8, INT32, INT32, False>,
std::tuple< Col, Col, Row, INT8, INT8, INT32, INT32, True>,
std::tuple< Col, Col, Row, INT8, INT8, INT32, INT32, False>,
std::tuple< Row, Row, Row, INT8, INT8, INT32, INT32, True>,
std::tuple< Row, Row, Row, INT8, INT8, INT32, INT32, False>,
std::tuple< Col, Row, Row, INT8, INT8, INT32, INT32, True>,
std::tuple< Col, Row, Row, INT8, INT8, INT32, INT32, False>
>;
// clang-format on