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

Merge commit 'b60af5bde965a2bb007bb582f7836b43ca647b81' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-30 16:14:10 +00:00
parent 631a25de61
commit ee9718a427
115 changed files with 2849 additions and 756 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/pool/test_max_pool3d_fwd.cpp
View File

@@ -5,6 +5,9 @@
#include "profiler/profile_pool3d_fwd_impl.hpp"
#include "test_pool_fwd_common.hpp"
static ck::index_t param_mask = 0xffff;
static ck::index_t instance_index = -1;
template <typename Tuple>
class TestMaxPool3dFwd : public ::testing::Test
{
@@ -21,8 +24,14 @@ class TestMaxPool3dFwd : public ::testing::Test
void Run()
{
for(auto param : params)
for(size_t i = 0; i < this->params.size(); i++)
{
if((param_mask & (1 << i)) == 0)
{
continue;
}
auto& param = this->params[i];
ck::profiler::PoolFwdKernelParams kernel_params{param.length_,
param.window_spatial_lengths_,
param.window_strides_,
@@ -40,7 +49,8 @@ class TestMaxPool3dFwd : public ::testing::Test
ck::tensor_layout::convolution::NDHWC,
ck::ReduceTensorOp::MAX,
false,
false>(in_params_max_pool, kernel_params);
false>(
in_params_max_pool, kernel_params, instance_index);
EXPECT_TRUE(success);
// max pool + index
@@ -52,8 +62,8 @@ class TestMaxPool3dFwd : public ::testing::Test
ck::tensor_layout::convolution::NDHWC,
ck::ReduceTensorOp::MAX,
false,
true>(in_params_max_pool_indexed,
kernel_params);
true>(
in_params_max_pool_indexed, kernel_params, instance_index);
EXPECT_TRUE(success);
}
}
@@ -76,3 +86,20 @@ TYPED_TEST(TestMaxPool3dFwd, Test_Pool)
this->Run();
}
int main(int argc, char** argv)
{
testing::InitGoogleTest(&argc, argv);
if(argc == 1) {}
else if(argc == 3)
{
param_mask = strtol(argv[1], nullptr, 0);
instance_index = atoi(argv[2]);
}
else
{
std::cout << "Usage of " << argv[0] << std::endl;
std::cout << "Arg1,2: param_mask instance_index(-1 means all)" << std::endl;
}
return RUN_ALL_TESTS();
}