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

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

batched_transpose_common_policy.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

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

3 

4 #pragma once

5 

6 #include "ck_tile/core.hpp"

7 

8 namespace ck_tile {

9 

10 struct BatchedTransposeCommonPolicy

11 {

12  CK_TILE_DEVICE static constexpr auto TileAccessPattern =

13  tile_distribution_pattern::thread_raked;

14 

15  template <typename Problem>

16  CK_TILE_DEVICE static constexpr auto MakeInputDistribution()

17  {

18  constexpr index_t kBlockSize = Problem::kBlockSize;

19  constexpr index_t kLeadDimPerBlock = Problem::kNPerBlock;

20  constexpr index_t kSecondDimPerBlock = Problem::kMPerBlock;

21 

22  constexpr index_t kVectorSize = Problem::VectorSizeInput;

23  static_assert((kLeadDimPerBlock * kVectorSize) % kBlockSize == 0, "");

24  using TileEncodingPattern = tile_distribution_encoding_pattern_2d<kBlockSize,

25  kSecondDimPerBlock,

26  kLeadDimPerBlock,

27  kVectorSize,

28  TileAccessPattern>;

29  return TileEncodingPattern::make_2d_static_tile_distribution();

30  }

31 };

32 

33 } // namespace ck_tile

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::tile_distribution_pattern::thread_raked

@ thread_raked

Thread raked pattern.

ck_tile::BatchedTransposeCommonPolicy

Definition: batched_transpose_common_policy.hpp:11

ck_tile::BatchedTransposeCommonPolicy::MakeInputDistribution

static constexpr CK_TILE_DEVICE auto MakeInputDistribution()

Definition: batched_transpose_common_policy.hpp:16

ck_tile::BatchedTransposeCommonPolicy::TileAccessPattern

static constexpr CK_TILE_DEVICE auto TileAccessPattern

Definition: batched_transpose_common_policy.hpp:12

ck_tile::tile_distribution_encoding_pattern_2d

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

Definition: static_encoding_pattern.hpp:130