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

• develop
• include
• ck_tile
• ops
• elementwise
• pipeline

elementwise_pipeline_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 

4 #pragma once

5 

6 #include "ck_tile/core/utility/type_traits.hpp"

7 

8 namespace ck_tile {

9 

10 template <typename XDataType_,

11  typename ComputeDataType_,

12  typename YDataType_,

13  typename BlockShape_,

14  typename ElementWiseOperation_,

15  bool kPad_ = true>

16 struct ElementWisePipelineProblem

17 {

18  using XDataType = remove_cvref_t<XDataType_>;

19  using ComputeDataType = remove_cvref_t<ComputeDataType_>;

20  using YDataType = remove_cvref_t<YDataType_>;

21  using BlockShape = remove_cvref_t<BlockShape_>;

22  using ElementWiseOperation = remove_cvref_t<ElementWiseOperation_>;

23  static constexpr bool kPad = kPad_;

24 };

25 

26 } // namespace ck_tile

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition: type_traits.hpp:21

ck_tile::ElementWisePipelineProblem

Definition: elementwise_pipeline_problem.hpp:17

ck_tile::ElementWisePipelineProblem::kPad

static constexpr bool kPad

Definition: elementwise_pipeline_problem.hpp:23

ck_tile::ElementWisePipelineProblem::ElementWiseOperation

remove_cvref_t< ElementWiseOperation_ > ElementWiseOperation

Definition: elementwise_pipeline_problem.hpp:22

ck_tile::ElementWisePipelineProblem::ComputeDataType

remove_cvref_t< ComputeDataType_ > ComputeDataType

Definition: elementwise_pipeline_problem.hpp:19

ck_tile::ElementWisePipelineProblem::XDataType

remove_cvref_t< XDataType_ > XDataType

Definition: elementwise_pipeline_problem.hpp:18

ck_tile::ElementWisePipelineProblem::YDataType

remove_cvref_t< YDataType_ > YDataType

Definition: elementwise_pipeline_problem.hpp:20

ck_tile::ElementWisePipelineProblem::BlockShape

remove_cvref_t< BlockShape_ > BlockShape

Definition: elementwise_pipeline_problem.hpp:21

type_traits.hpp