/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/arch/mma/wmma/wmma_traits.hpp Source File

• develop
• include
• ck_tile
• core
• arch
• mma
• wmma

wmma_traits.hpp

Go to the documentation of this file.

1 // Copyright (c) Advanced Micro Devices, Inc., or its affiliates.

2 // SPDX-License-Identifier: MIT

3 

4 #pragma once

5 

6 namespace ck_tile::core::arch::mma {

7 

13 struct WmmaOp;

14 

20 template <typename MmaOp, typename = void>

21 struct is_mma_op_wmma : std::false_type

22 {

23 };

24 

30 template <typename MmaOp>

31 // TODO: c++20 requires

32 struct is_mma_op_wmma<MmaOp, std::enable_if_t<std::is_same_v<typename MmaOp::OpType, WmmaOp>>>

33  : std::true_type

34 {

35 };

36 

41 template <typename MmaOp>

42 static constexpr bool is_mma_op_wmma_v = is_mma_op_wmma<MmaOp>::value;

43 

44 } // namespace ck_tile::core::arch::mma

ck_tile::core::arch::mma

Definition: amdgcn_mma.hpp:10

ck::false_type

bool_constant< false > false_type

Definition: integral_constant.hpp:63

ck::enable_if_t

typename std::enable_if< B, T >::type enable_if_t

Definition: enable_if.hpp:27

ck::true_type

bool_constant< true > true_type

Definition: integral_constant.hpp:62

WmmaOp

Meta-tag for the WMMA operation. This will be used in the MmaOp struct to identify the operation as a...

ck_tile::core::arch::mma::is_mma_op_wmma

Trait to check if MmaOp is an WMMA operation.

Definition: wmma_traits.hpp:22