include/ck_tile/ops/gemm_group_quant/pipeline/tile_gemm_aquant_traits.hpp Source File

• include
• ck_tile
• ops
• gemm_group_quant
• pipeline

tile_gemm_aquant_traits.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2025, 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 template <bool kPadM_,

11  bool kPadN_,

12  bool kPadK_,

13  typename ALayout_,

14  typename BLayout_,

15  typename CLayout_,

16  typename AQLayout_ = ALayout_>

17 struct TileGemmAQuantTraits

18 {

19  static constexpr bool kPadM = kPadM_;

20  static constexpr bool kPadN = kPadN_;

21  static constexpr bool kPadK = kPadK_;

22 

23  static constexpr int _VectorSize = 16;

24 

25  using ALayout = ALayout_;

26  using BLayout = BLayout_;

27  using CLayout = CLayout_;

28  using AQLayout = AQLayout_;

29 

30  static constexpr bool UseStructuredSparsity = false;

31  static constexpr index_t NumWaveGroups = 1;

32 };

33 

34 } // namespace ck_tile

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::index_t

int32_t index_t

Definition: integer.hpp:9

ck_tile::TileGemmAQuantTraits

Definition: tile_gemm_aquant_traits.hpp:18

ck_tile::TileGemmAQuantTraits::NumWaveGroups

static constexpr index_t NumWaveGroups

Definition: tile_gemm_aquant_traits.hpp:31

ck_tile::TileGemmAQuantTraits::kPadN

static constexpr bool kPadN

Definition: tile_gemm_aquant_traits.hpp:20

ck_tile::TileGemmAQuantTraits::kPadM

static constexpr bool kPadM

Definition: tile_gemm_aquant_traits.hpp:19

ck_tile::TileGemmAQuantTraits::UseStructuredSparsity

static constexpr bool UseStructuredSparsity

Definition: tile_gemm_aquant_traits.hpp:30

ck_tile::TileGemmAQuantTraits::kPadK

static constexpr bool kPadK

Definition: tile_gemm_aquant_traits.hpp:21

ck_tile::TileGemmAQuantTraits::CLayout

CLayout_ CLayout

Definition: tile_gemm_aquant_traits.hpp:27

ck_tile::TileGemmAQuantTraits::AQLayout

AQLayout_ AQLayout

Definition: tile_gemm_aquant_traits.hpp:28

ck_tile::TileGemmAQuantTraits::ALayout

ALayout_ ALayout

Definition: tile_gemm_aquant_traits.hpp:25

ck_tile::TileGemmAQuantTraits::_VectorSize

static constexpr int _VectorSize

Definition: tile_gemm_aquant_traits.hpp:23

ck_tile::TileGemmAQuantTraits::BLayout

BLayout_ BLayout

Definition: tile_gemm_aquant_traits.hpp:26