include/ck/tensor_operation/gpu/device/device_grouped_gemm_fixed_nk.hpp Source File

• include
• ck
• tensor_operation
• gpu
• device

device_grouped_gemm_fixed_nk.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 "device_grouped_gemm_splitk.hpp"

7 

8 namespace ck {

9 namespace tensor_operation {

10 namespace device {

11 

12 template <typename ALayout,

13  typename BLayout,

14  typename DsLayout,

15  typename ELayout,

16  typename ADataType,

17  typename BDataType,

18  typename DsDataType,

19  typename EDataType,

20  typename AElementwiseOperation,

21  typename BElementwiseOperation,

22  typename CElementwiseOperation>

23 struct DeviceGroupedGemmFixedNK : DeviceGroupedGemmSplitK<ALayout,

24  BLayout,

25  DsLayout,

26  ELayout,

27  ADataType,

28  BDataType,

29  DsDataType,

30  EDataType,

31  AElementwiseOperation,

32  BElementwiseOperation,

33  CElementwiseOperation>

34 {

35 };

36 

37 } // namespace device

38 } // namespace tensor_operation

39 } // namespace ck

device_grouped_gemm_splitk.hpp

ck

Definition: ck.hpp:264

ck::tensor_operation::device::DeviceGroupedGemmFixedNK

Definition: device_grouped_gemm_fixed_nk.hpp:34

ck::tensor_operation::device::DeviceGroupedGemmSplitK

Definition: device_grouped_gemm_splitk.hpp:33