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

• develop
• include
• ck_tile
• ops
• batched_contraction
• pipeline

batched_contraction_problem.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 #pragma once

4 

5 #include "ck_tile/core.hpp"

6 

7 namespace ck_tile {

8 

9 template <typename ADataType_,

10  typename BDataType_,

11  typename DsDataType_,

12  typename EDataType_,

13  ck_tile::index_t NumDimG_,

14  ck_tile::index_t NumDimM_,

15  ck_tile::index_t NumDimN_,

16  ck_tile::index_t NumDimK_,

17  ck_tile::index_t NumDTensor_>

18 struct BatchedContractionProblem

19 {

20  using ADataType = ck_tile::remove_cvref_t<ADataType_>;

21  using BDataType = ck_tile::remove_cvref_t<BDataType_>;

22  using DsDataType = ck_tile::remove_cvref_t<DsDataType_>;

23  using EDataType = ck_tile::remove_cvref_t<EDataType_>;

24 

25  static constexpr ck_tile::index_t NumDimG = NumDimG_;

26  static constexpr ck_tile::index_t NumDimM = NumDimM_;

27  static constexpr ck_tile::index_t NumDimN = NumDimN_;

28  static constexpr ck_tile::index_t NumDimK = NumDimK_;

29  static constexpr ck_tile::index_t NumDTensor = NumDTensor_;

30 };

31 

32 } // 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::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition: type_traits.hpp:21

ck_tile::BatchedContractionProblem

Definition: batched_contraction_problem.hpp:19

ck_tile::BatchedContractionProblem::ADataType

ck_tile::remove_cvref_t< ADataType_ > ADataType

Definition: batched_contraction_problem.hpp:20

ck_tile::BatchedContractionProblem::EDataType

ck_tile::remove_cvref_t< EDataType_ > EDataType

Definition: batched_contraction_problem.hpp:23

ck_tile::BatchedContractionProblem::DsDataType

ck_tile::remove_cvref_t< DsDataType_ > DsDataType

Definition: batched_contraction_problem.hpp:22

ck_tile::BatchedContractionProblem::NumDimK

static constexpr ck_tile::index_t NumDimK

Definition: batched_contraction_problem.hpp:28

ck_tile::BatchedContractionProblem::NumDTensor

static constexpr ck_tile::index_t NumDTensor

Definition: batched_contraction_problem.hpp:29

ck_tile::BatchedContractionProblem::NumDimM

static constexpr ck_tile::index_t NumDimM

Definition: batched_contraction_problem.hpp:26

ck_tile::BatchedContractionProblem::NumDimN

static constexpr ck_tile::index_t NumDimN

Definition: batched_contraction_problem.hpp:27

ck_tile::BatchedContractionProblem::BDataType

ck_tile::remove_cvref_t< BDataType_ > BDataType

Definition: batched_contraction_problem.hpp:21

ck_tile::BatchedContractionProblem::NumDimG

static constexpr ck_tile::index_t NumDimG

Definition: batched_contraction_problem.hpp:25