mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 05:01:25 +00:00
f60f0a5e03
* Do not hardcode stride * devicePool2DFwd Inherit devicePool3DFwd * Move instance declaration out of common * Add dilation * use the pool3d rank, because pool2d inherit pooo3d * calculate Do Ho Wo for the dilation * Fix header name * Modify ckProfiler * Remove pool2d instance * Remove pool2d in profiler * Remove pool2d and add dilation * In to client example, this commit revise following: 1. Add dilation. 2. Use pool3d to implement pool2d * Refine naming and IsSupportedArgument() * Add dilation to maxpool bwd example * clang format * 1. Remove useless header 2. Fix copyright 3. Refine naming * Add layout parameter to pool fwd * clang format * Fix merge error * Fix compile error * Remove layout parameter in derived class * Refine changlog * Fix compile error * Fix compiler error * Add layout to external api and profiler
123 lines
4.1 KiB
C++
123 lines
4.1 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include <iostream>
|
|
|
|
#include "ck/ck.hpp"
|
|
#include "ck/utility/reduction_enums.hpp"
|
|
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
|
|
|
|
#include "pool2d_fwd_common.hpp"
|
|
|
|
using InDataType = float;
|
|
using OutDataType = float;
|
|
using ComputeDataType = float;
|
|
|
|
using IndexDataType = int32_t;
|
|
|
|
using InLayout = ck::tensor_layout::convolution::NHWC;
|
|
using OutLayout = ck::tensor_layout::convolution::NHWC;
|
|
|
|
#if 1
|
|
static constexpr auto ReduceOpId = ck::ReduceTensorOp::MAX;
|
|
#else
|
|
static constexpr auto ReduceOpId = ck::ReduceTensorOp::AVG;
|
|
#endif
|
|
|
|
static constexpr bool OutputIndex = false;
|
|
static constexpr bool PropagateNan = false;
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
bool do_verification;
|
|
int init_method;
|
|
bool time_kernel;
|
|
|
|
// Pool shape
|
|
ck::index_t N = 128;
|
|
ck::index_t C = 192;
|
|
ck::index_t Y = 3;
|
|
ck::index_t X = 3;
|
|
ck::index_t Hi = 71;
|
|
ck::index_t Wi = 71;
|
|
ck::index_t window_stride_h = 2;
|
|
ck::index_t window_stride_w = 2;
|
|
ck::index_t window_dilation_h = 1;
|
|
ck::index_t window_dilation_w = 1;
|
|
ck::index_t in_left_pad_h = 1;
|
|
ck::index_t in_left_pad_w = 1;
|
|
ck::index_t in_right_pad_h = 1;
|
|
ck::index_t in_right_pad_w = 1;
|
|
|
|
if(argc == 1)
|
|
{
|
|
do_verification = true;
|
|
init_method = 1;
|
|
time_kernel = true;
|
|
}
|
|
else if(argc == 4)
|
|
{
|
|
do_verification = std::stoi(argv[1]);
|
|
init_method = std::stoi(argv[2]);
|
|
time_kernel = static_cast<bool>(std::stoi(argv[3]));
|
|
}
|
|
else if(argc == 18)
|
|
{
|
|
do_verification = std::stoi(argv[1]);
|
|
init_method = std::stoi(argv[2]);
|
|
time_kernel = static_cast<bool>(std::stoi(argv[3]));
|
|
|
|
N = std::stoi(argv[4]);
|
|
C = std::stoi(argv[5]);
|
|
Y = std::stoi(argv[6]);
|
|
X = std::stoi(argv[7]);
|
|
Hi = std::stoi(argv[8]);
|
|
Wi = std::stoi(argv[9]);
|
|
window_stride_h = std::stoi(argv[10]);
|
|
window_stride_w = std::stoi(argv[11]);
|
|
window_dilation_h = std::stoi(argv[12]);
|
|
window_dilation_w = std::stoi(argv[13]);
|
|
in_left_pad_h = std::stoi(argv[14]);
|
|
in_left_pad_w = std::stoi(argv[15]);
|
|
in_right_pad_h = std::stoi(argv[16]);
|
|
in_right_pad_w = std::stoi(argv[17]);
|
|
}
|
|
else
|
|
{
|
|
printf("arg1: verification (0=no, 1=yes)\n");
|
|
printf("arg2: initialization (0=no init, 1=integer value, 2=decimal value)\n");
|
|
printf("arg3: time kernel (0=no, 1=yes)\n");
|
|
printf("arg4 to 15: N, C, Y, X, Hi, Wi, Sy, Sx, Dy, Dx, LeftPy, LeftPx, RightPy, "
|
|
"RightPx\n");
|
|
exit(0);
|
|
}
|
|
|
|
bool pass = pool_test<InDataType,
|
|
OutDataType,
|
|
ComputeDataType,
|
|
IndexDataType,
|
|
InLayout,
|
|
OutLayout,
|
|
ReduceOpId,
|
|
PropagateNan,
|
|
OutputIndex>(do_verification,
|
|
init_method,
|
|
time_kernel,
|
|
N,
|
|
C,
|
|
Y,
|
|
X,
|
|
Hi,
|
|
Wi,
|
|
window_stride_h,
|
|
window_stride_w,
|
|
window_dilation_h,
|
|
window_dilation_w,
|
|
in_left_pad_h,
|
|
in_left_pad_w,
|
|
in_right_pad_h,
|
|
in_right_pad_w);
|
|
|
|
return (pass ? 0 : 1);
|
|
}
|