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

Merge commit '91e32f305fa4d809103431a81594c52240753d40' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2026-01-27 22:14:22 +00:00
parent 362767eba8
commit d4b61e4db5
87 changed files with 9113 additions and 20144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/ck_tile/reduce/test_multi_reduce2d_threadwise.cpp
View File

@@ -39,26 +39,20 @@ using TestConfig_F16_Add = std::tuple<ck_tile::half_t,
Shape1_WarpTile,
Shape1_ThreadTile>;
using TestConfig_F16_Add_Max = std::tuple<
using TestConfig_F16_Add_SumSquare = std::tuple<
ck_tile::half_t,
float,
ck_tile::half_t,
ck_tile::tuple<ck_tile::ReduceOp::Add, ck_tile::ReduceOp::Max, ck_tile::ReduceOp::Add>,
ck_tile::tuple<ck_tile::element_wise::PassThrough,
ck_tile::element_wise::PassThrough,
ck_tile::element_wise::UnarySquare>,
ck_tile::tuple<ck_tile::element_wise::PassThrough,
ck_tile::element_wise::PassThrough,
ck_tile::element_wise::UnaryDivide>,
ck_tile::tuple<ck_tile::element_wise::PassThrough,
ck_tile::element_wise::PassThrough,
ck_tile::element_wise::PassThrough>,
ck_tile::tuple<ck_tile::ReduceOp::Add, ck_tile::ReduceOp::Add>,
ck_tile::tuple<ck_tile::element_wise::PassThrough, ck_tile::element_wise::UnarySquare>,
ck_tile::tuple<ck_tile::element_wise::PassThrough, ck_tile::element_wise::UnaryDivide>,
ck_tile::tuple<ck_tile::element_wise::PassThrough, ck_tile::element_wise::PassThrough>,
Shape1_BlockWarps,
Shape1_BlockTile,
Shape1_WarpTile,
Shape1_ThreadTile>;
using TestTypes = ::testing::Types<TestConfig_F16_Add, TestConfig_F16_Add_Max>;
using TestTypes = ::testing::Types<TestConfig_F16_Add, TestConfig_F16_Add_SumSquare>;
TYPED_TEST_SUITE(TestCkTileMultiReduceThreadwise, TestTypes);