mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-29 11:16:59 +00:00
717f2efef7
[CK] add composable kernel support on gfx1250 (#6978) ## Motivation Add composable kernel support on gfx1250. ## Technical Details <!-- Explain the changes along with any relevant GitHub links. --> ## Test Plan <!-- Explain any relevant testing done to verify this PR. --> ## Test Result <!-- Briefly summarize test outcomes. --> ## Submission Checklist - [ ] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests. --------- Co-authored-by: Qun Lin <qlin@amd.com> Co-authored-by: jialuo12_amdeng <jia.luo@amd.com> Co-authored-by: Andriy Roshchenko <andriy.roshchenko@amd.com> Co-authored-by: hsivasun_amdeng <haresh.sivasuntharampillai@amd.com>
67 lines
2.3 KiB
C++
67 lines
2.3 KiB
C++
// Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#include "common.hpp"
|
|
|
|
#include "ck/tensor_operation/gpu/device/impl/device_gemm_wmma_cshuffle_v3.hpp"
|
|
|
|
using ADataType = ck::f8_t;
|
|
using BDataType = ck::f8_t;
|
|
using AccDataType = float;
|
|
using CShuffleDataType = ck::bhalf_t;
|
|
using CDataType = ck::bhalf_t;
|
|
using ComputeTypeA = ck::f8_t;
|
|
using ComputeTypeB = ck::f8_t;
|
|
|
|
using ALayout = Col;
|
|
using BLayout = Col;
|
|
using CLayout = Row;
|
|
|
|
using AElementOp = PassThrough;
|
|
using BElementOp = PassThrough;
|
|
using CElementOp = PassThrough;
|
|
|
|
static constexpr auto GemmDefault = ck::tensor_operation::device::GemmSpecialization::Default;
|
|
|
|
// clang-format off
|
|
using DeviceGemmV2Instance = ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3<
|
|
ALayout, BLayout, CLayout,
|
|
ADataType, BDataType, CDataType, AccDataType, CShuffleDataType,
|
|
PassThrough, PassThrough, PassThrough, GemmDefault,
|
|
128,
|
|
128, 64, 128,
|
|
16, 16, // AK1, BK1
|
|
16, 16,
|
|
4, 2,
|
|
S<4, 32, 1>, S<0, 2, 1>, S<0, 2, 1>,
|
|
1, 4, 16, 0,
|
|
S<4, 32, 1>, S<1, 0, 2>, S<1, 0, 2>,
|
|
2, 16, 16, 0,
|
|
1, 1, S<1, 32, 1, 4>, 8,
|
|
ck::BlockGemmPipelineScheduler::Intrawave, ck::BlockGemmPipelineVersion::v1,
|
|
ComputeTypeA, ComputeTypeB>;
|
|
// clang-format on
|
|
|
|
using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataType,
|
|
BDataType,
|
|
CDataType,
|
|
AccDataType,
|
|
AElementOp,
|
|
BElementOp,
|
|
CElementOp,
|
|
ComputeTypeA,
|
|
ComputeTypeB>;
|
|
|
|
#include "run_gemm_example_v2.inc"
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
if(!ck::is_gfx12_supported())
|
|
{
|
|
std::cout << "This kernel support gfx12 only" << std::endl;
|
|
|
|
return 0;
|
|
}
|
|
return !run_gemm_splitk_example(argc, argv);
|
|
}
|