mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
e9fd122889
* Conv3d bwd weight client example. * Update year in license * Convolution bwd data 3D fp16/fp32 client example. * Client example for convnd fwd fp16 fp32 * clang-format * Review remarks. * Fix compiler err. * Update data layout to standard one. * Add conv 3d fwd NDHWGC instances * clang-format * Conv3d fwd NDHWGC instances. --------- Co-authored-by: Adam Osewski <aosewski@amd.com> Co-authored-by: zjing14 <zhangjing14@gmail.com>
45 lines
1.6 KiB
C++
45 lines
1.6 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "common.hpp"
|
|
|
|
#include "ck/ck.hpp"
|
|
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
|
|
|
|
using InDataType = ck::half_t;
|
|
using WeiDataType = ck::half_t;
|
|
using OutDataType = ck::half_t;
|
|
|
|
using InLayout = ck::tensor_layout::convolution::NDHWGC;
|
|
using WeiLayout = ck::tensor_layout::convolution::KZYXGC;
|
|
using OutLayout = ck::tensor_layout::convolution::NDHWGK;
|
|
|
|
static constexpr ck::index_t NumDimSpatial = 3;
|
|
static constexpr ck::index_t G = 1;
|
|
static constexpr ck::index_t N = 64;
|
|
static constexpr ck::index_t K = 128;
|
|
static constexpr ck::index_t C = 64;
|
|
static constexpr ck::index_t Z = 3;
|
|
static constexpr ck::index_t Y = 3;
|
|
static constexpr ck::index_t X = 3;
|
|
static constexpr ck::index_t Di = 28;
|
|
static constexpr ck::index_t Hi = 28;
|
|
static constexpr ck::index_t Wi = 3;
|
|
static constexpr ck::index_t Do = 28;
|
|
static constexpr ck::index_t Ho = 28;
|
|
static constexpr ck::index_t Wo = 3;
|
|
|
|
int main()
|
|
{
|
|
return run_grouped_conv_fwd<NumDimSpatial,
|
|
InDataType,
|
|
WeiDataType,
|
|
OutDataType,
|
|
InLayout,
|
|
WeiLayout,
|
|
OutLayout>(
|
|
{N, Di, Hi, Wi, G, C}, {K, Z, Y, X, G, C}, {N, Do, Ho, Wo, G, K})
|
|
? EXIT_SUCCESS
|
|
: EXIT_FAILURE;
|
|
}
|