mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-29 03:07:02 +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>
29 lines
874 B
C++
29 lines
874 B
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "common.hpp"
|
|
#include "ck/library/reference_tensor_operation/cpu/reference_mx_gemm.hpp"
|
|
|
|
#include "mx_wp_gemm_xdl_benchmark_instance.hpp"
|
|
|
|
#include "run_mx_gemm_example_v3.inc"
|
|
|
|
namespace ck::tensor_operation::device::instance {
|
|
|
|
extern void add_mx_wp_gemm_xdl_benchmark_instances(mx_wp_gemm_xdl_benchmark_instances& instances);
|
|
|
|
bool init_opt_ptrs(mx_wp_gemm_xdl_benchmark_instances& op_ptrs)
|
|
{
|
|
add_mx_wp_gemm_xdl_benchmark_instances(op_ptrs);
|
|
return true;
|
|
}
|
|
|
|
} // namespace ck::tensor_operation::device::instance
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
mx_wp_gemm_xdl_benchmark_instances op_ptrs;
|
|
return !ck::tensor_operation::device::instance::init_opt_ptrs(op_ptrs) ||
|
|
!run_mx_gemm_splitk_example<true>(op_ptrs, argc, argv);
|
|
}
|