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

• develop
• include
• ck_tile
• core
• arch
• mma
• mfma

mfma_transforms.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 #include "ck_tile/core/arch/arch.hpp"

7 

8 namespace ck_tile::core::arch::mma {

9 

14 struct MmaDefaultTransformsGfx9

15 {

16  using ATransform = PassThroughTransform;

17  using BTransform = PassThroughTransform;

18  using CTransform = PassThroughTransform;

19  using DTransform = PassThroughTransform;

20 };

21 

28 // TODO: c++20 template <MmaOpI MmaOp, amdgcn_target_arch_id CompilerTarget>

29 // TODO: c++20 requires

30 template <typename MmaOp, typename CompilerTarget>

31 struct MmaTransformsDefaultSelector<MmaOp,

32  CompilerTarget,

33  enable_if_target_family_gfx9_t<CompilerTarget>>

34 {

35  using SelectedTransforms = MmaDefaultTransformsGfx9;

36 };

37 

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

arch.hpp

ck_tile::core::arch::mma

Definition: amdgcn_mma.hpp:10

ck_tile::core::arch::mma::MmaDefaultTransformsGfx9

Implements the default MMA transforms for gfx9 targets.

Definition: mfma_transforms.hpp:15

ck_tile::core::arch::mma::MmaTransformsDefaultSelector

Implements the default MMA transforms selection for gfx9 targets.

Definition: mma_transforms.hpp:28

ck_tile::core::arch::mma::PassThroughTransform

A no-op transform that passes through the input as-is.

Definition: mma_transforms.hpp:11