device_grouped_gemm_fixed_nk.hpp Source File

device_grouped_gemm_fixed_nk.hpp Source File#

Composable Kernel: device_grouped_gemm_fixed_nk.hpp Source File
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
8namespace ck {
9namespace tensor_operation {
10namespace device {
11
12template <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>
23struct 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::tensor_operation::device
Definition convolution_backward_data_specialization.hpp:8
ck::tensor_operation
Definition convolution_backward_data_specialization.hpp:7
ck
Definition ck.hpp:268
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