test_transpose.cpp 661 Bytes
Newer Older
Astha Rai's avatar
Astha Rai committed
1
2
3
4
5
6
7
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.

#include <tuple>

#include "gtest/gtest.h"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
Astha Rai's avatar
Astha Rai committed
8
#include "test_transpose_util.hpp"
Astha Rai's avatar
Astha Rai committed
9
10
11
12
13

using F16 = ck::half_t;
using F32 = float;

template <typename Tuple>
Astha Rai's avatar
Astha Rai committed
14
class TestTranspose : public ck::test::TestTranspose<Tuple>
Astha Rai's avatar
Astha Rai committed
15
16
17
18
19
20
21
22
23
24
{
};

// clang-format off
using KernelTypes = ::testing::Types<
    std::tuple<      F16,       F16>,
    std::tuple<      F32,       F32>
    >;
// clang-format on

Astha Rai's avatar
Astha Rai committed
25
TYPED_TEST_SUITE(TestTranspose, KernelTypes);
Astha Rai's avatar
Astha Rai committed
26

Astha Rai's avatar
Astha Rai committed
27
//#include "test_transpose_ut_cases.inc"