/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_transpose/pipeline/batched_transpose_policy.hpp Source File

• develop
• include
• ck_tile
• ops
• batched_transpose
• pipeline

batched_transpose_policy.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 #include "ck_tile/core.hpp"

7 #include "batched_transpose_common_policy.hpp"

8 

9 namespace ck_tile {

10 

11 struct BatchedTransposePolicy : public BatchedTransposeCommonPolicy

12 {

13  template <typename Problem>

14  CK_TILE_DEVICE static constexpr auto MakeOutputDistribution()

15  {

16  constexpr index_t BlockSize = Problem::kBlockSize;

17  constexpr index_t MPerBlock = Problem::kMPerBlock;

18  constexpr index_t NPerBlock = Problem::kNPerBlock;

19  constexpr index_t VecLoadSize = Problem::VectorSizeOutput;

20 

21  using TileEncodingPattern = tile_distribution_encoding_pattern_2d<BlockSize,

22  MPerBlock,

23  NPerBlock,

24  VecLoadSize,

25  TileAccessPattern>;

26  return TileEncodingPattern::make_shuffled_2d_static_tile_distribution();

27  }

28 };

29 } // namespace ck_tile

batched_transpose_common_policy.hpp

CK_TILE_DEVICE

#define CK_TILE_DEVICE

Definition: config.hpp:41

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::index_t

int32_t index_t

Definition: integer.hpp:9

ck_tile::BatchedTransposeCommonPolicy

Definition: batched_transpose_common_policy.hpp:11

ck_tile::BatchedTransposeCommonPolicy::TileAccessPattern

static constexpr CK_TILE_DEVICE auto TileAccessPattern

Definition: batched_transpose_common_policy.hpp:12

ck_tile::BatchedTransposePolicy

Definition: batched_transpose_policy.hpp:12

ck_tile::BatchedTransposePolicy::MakeOutputDistribution

static constexpr CK_TILE_DEVICE auto MakeOutputDistribution()

Definition: batched_transpose_policy.hpp:14

ck_tile::tile_distribution_encoding_pattern_2d

Class creating 2D static tile distribution with different load/store patterns.

Definition: static_encoding_pattern.hpp:130