mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 08:50:17 +00:00
76ec0089fb
* Expand the base class of pool2d, prepare to share base class with pool3d * Add pool3d device op * Add pool3d f16 example * Refactor the base class. implement generic pooling in the future * clang format * get original index in max pooling * Add outputindex to base class * Fix dimension * Add pooling instance * Use indexType instead * Remove useless header * Extract IndexDataType to template * Extract pooling reference code * clang format * clang format * Fix typo * Add tensor stride * Add missing header * Add index stride and output stride * Refine naming * Add type to base class * Rename file * Use proper size * Fix typo * Refine naming * Modify the argument into vector. * Add max pool profiler * Refine naming * Support f32 pool * Fix typo * Add avg pool2d fwd in profiler * clang format * Rename AccDatatype to ComputeDatatype * Fix init * test pool * Extract variable * Add client example * Check the pooling dim * clang format * Connect argv and arg_parser * Add found check * Remove useless header * Refine naming * Adjust the order of device_pool_fwd
32 lines
1020 B
C++
32 lines
1020 B
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "ck/ck.hpp"
|
|
|
|
using F16 = ck::half_t;
|
|
using F32 = float;
|
|
using I32 = int32_t;
|
|
using ck::index_t;
|
|
|
|
struct PoolingParam
|
|
{
|
|
PoolingParam(const std::vector<index_t>& length,
|
|
const std::vector<index_t>& window_spatial_lengths,
|
|
const std::vector<index_t>& window_strides,
|
|
const std::vector<index_t>& input_left_pads,
|
|
const std::vector<index_t>& input_right_pads)
|
|
: length_(length),
|
|
window_spatial_lengths_(window_spatial_lengths),
|
|
window_strides_(window_strides),
|
|
input_left_pads_(input_left_pads),
|
|
input_right_pads_(input_right_pads)
|
|
{
|
|
}
|
|
std::vector<index_t> length_;
|
|
std::vector<index_t> window_spatial_lengths_;
|
|
std::vector<index_t> window_strides_;
|
|
std::vector<index_t> input_left_pads_;
|
|
std::vector<index_t> input_right_pads_;
|
|
};
|