mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-05 22:22:27 +00:00
2ea75bd6d7
* split the types in gemm_bilinear instances, add condition to cmake policy * fix syntax * split the data types in batchnorm examples * fix the batchnorm_bwd test * fix types in the batchnorm_bwd test
106 lines
3.7 KiB
C++
106 lines
3.7 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <initializer_list>
|
|
#include <vector>
|
|
#include <tuple>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "profiler/profile_batchnorm_backward_impl.hpp"
|
|
|
|
using F16 = ck::half_t;
|
|
using F32 = float;
|
|
using BF16 = ck::bhalf_t;
|
|
using F64 = double;
|
|
|
|
template <typename Tuple>
|
|
class TestBatchNormBwdRank4 : public ::testing::Test
|
|
{
|
|
private:
|
|
const double epsilon = std::numeric_limits<float>::epsilon();
|
|
|
|
protected:
|
|
using XDataType = std::tuple_element_t<0, Tuple>;
|
|
using DxDataType = std::tuple_element_t<1, Tuple>;
|
|
using DyDataType = std::tuple_element_t<2, Tuple>;
|
|
using AccDataType = std::tuple_element_t<3, Tuple>;
|
|
using ScaleDataType = std::tuple_element_t<4, Tuple>;
|
|
using BiasDataType = std::tuple_element_t<5, Tuple>;
|
|
using MeanVarDataType = std::tuple_element_t<6, Tuple>;
|
|
|
|
std::vector<std::vector<size_t>> list_of_lengths = {
|
|
{128, 16, 3, 1024}, {128, 16, 6, 512}, {1, 1, 1, 1}, {4, 4, 4, 4}, {32, 32, 32, 32}};
|
|
std::vector<int> reduceDims;
|
|
|
|
template <int NumReduceDim>
|
|
void Run()
|
|
{
|
|
for(auto& inOutLengths : list_of_lengths)
|
|
{
|
|
bool pass = true;
|
|
|
|
EXPECT_FALSE(reduceDims.size() != NumReduceDim);
|
|
|
|
pass = pass && ck::profiler::profile_batchnorm_backward_impl<XDataType,
|
|
DxDataType,
|
|
DyDataType,
|
|
AccDataType,
|
|
ScaleDataType,
|
|
BiasDataType,
|
|
MeanVarDataType,
|
|
4,
|
|
NumReduceDim>(
|
|
true, 3, false, false, inOutLengths, reduceDims, true, epsilon);
|
|
|
|
pass = pass && ck::profiler::profile_batchnorm_backward_impl<XDataType,
|
|
DxDataType,
|
|
DyDataType,
|
|
AccDataType,
|
|
ScaleDataType,
|
|
BiasDataType,
|
|
MeanVarDataType,
|
|
4,
|
|
NumReduceDim>(
|
|
true, 3, false, false, inOutLengths, reduceDims, false, epsilon);
|
|
|
|
EXPECT_TRUE(pass);
|
|
}
|
|
}
|
|
};
|
|
|
|
using KernelTypes = ::testing::Types<
|
|
#ifdef CK_ENABLE_FP16
|
|
std::tuple<F16, F32, F32, F32, F16, F32, F32>
|
|
#endif
|
|
#ifdef CK_ENABLE_FP32
|
|
,
|
|
std::tuple<F32, F32, F32, F32, F32, F32, F32>
|
|
#endif
|
|
#ifdef CK_ENABLE_BF16
|
|
,
|
|
std::tuple<BF16, F32, F32, F32, BF16, F32, F32>
|
|
#endif
|
|
#ifdef CK_ENABLE_FP64
|
|
,
|
|
std::tuple<F64, F64, F64, F64, F64, F64, F64>
|
|
#endif
|
|
>;
|
|
|
|
TYPED_TEST_SUITE(TestBatchNormBwdRank4, KernelTypes);
|
|
|
|
// nhwc
|
|
TYPED_TEST(TestBatchNormBwdRank4, nhwc)
|
|
{
|
|
this->reduceDims = {0, 1, 2};
|
|
this->template Run<3>();
|
|
}
|
|
|
|
// nchw
|
|
TYPED_TEST(TestBatchNormBwdRank4, nchw)
|
|
{
|
|
this->reduceDims = {0, 2, 3};
|
|
this->template Run<3>();
|
|
}
|