mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 05:01:25 +00:00
578142db3a
* Add avgpool bwd reference code * Refine naming * Fix invalid in_element op in ref_conv * Add example (only reference now) * Add the full example of avgpool bwd * Fix copyright * Imitate MakeDescriptor from transform_conv_bwd_data_to_gemm_v1.hpp * rename channel to c from k * Arrange the code * Imitate the argument from conv bwd * Implement invoker * Fix order of parameter in example * Refactor reference code for different dimension * Support different stride * Check if argument is valid * Fix kernel parameter for NDHWC, fastest dimension C is not reduced * Add more data type in example * Fix bug in example * calculate Do Ho Wo according to the dilation * Remove useless header * Add comment in reference code * Add layout parameter * Remove layout in derived class * Refine reference comment
63 lines
2.2 KiB
C++
63 lines
2.2 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/tensor_operation/gpu/device/tensor_layout.hpp"
|
|
#include "ck/tensor_operation/gpu/device/impl/device_avgpool3d_bwd_ndhwc_ndhwc.hpp"
|
|
|
|
#include "avgpool3d_bwd_common.hpp"
|
|
|
|
using DOutDataType = ck::bhalf_t;
|
|
using DInDataType = ck::bhalf_t;
|
|
using ComputeDataType = float;
|
|
|
|
#if 1
|
|
using DOutLayout = ck::tensor_layout::convolution::NDHWC;
|
|
using DInLayout = ck::tensor_layout::convolution::NDHWC;
|
|
#else
|
|
using DOutLayout = ck::tensor_layout::convolution::NCDHW;
|
|
using DInLayout = ck::tensor_layout::convolution::NCDHW;
|
|
#endif
|
|
|
|
using DevicePoolBwdInstance =
|
|
ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC<DOutDataType,
|
|
DInDataType,
|
|
ComputeDataType,
|
|
64, // BlockSize
|
|
64, // ReduceMThreadClusterSize
|
|
1, // ReduceKThreadClusterSize
|
|
1, // ReduceMThreadSliceSize
|
|
1, // ReduceKThreadSliceSize
|
|
1>; // InSrcOutDstVectorSize
|
|
|
|
int main()
|
|
{
|
|
std::vector<ck::index_t> window_lengths = {5, 5, 5};
|
|
std::vector<ck::index_t> window_strides = {2, 2, 2};
|
|
std::vector<ck::index_t> window_dilations = {2, 2, 2};
|
|
std::vector<ck::index_t> dinput_left_pads = {0, 0, 0};
|
|
std::vector<ck::index_t> dinput_right_pads = {0, 0, 0};
|
|
|
|
ck::index_t N = 1;
|
|
ck::index_t C = 16;
|
|
ck::index_t Di = 40;
|
|
ck::index_t Hi = 40;
|
|
ck::index_t Wi = 40;
|
|
|
|
pool3d_bwd_test<DevicePoolBwdInstance, DOutDataType, DInDataType, DOutLayout, DInLayout>(
|
|
true,
|
|
false,
|
|
N,
|
|
C,
|
|
Di,
|
|
Hi,
|
|
Wi,
|
|
window_lengths,
|
|
window_strides,
|
|
window_dilations,
|
|
dinput_left_pads,
|
|
dinput_right_pads);
|
|
}
|