mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-06 15:54:31 +00:00
8eff4d62b6
* Format * Format * Format * Remove const * Use the right template * Format * Format * add row/col instances * Add missing file * fixed * Format * Updates * Format * fixed rrr layout * Format * Update test and embed modules * Restore older version * Update year * Set -fPIC * Format * Use double for isnan * rename host folder to codegen + minor fix * add codegen CI test * add option to build components without building CK * fix the groovy syntax * fix typo * use the correct function for the codegen stage --------- Co-authored-by: Jing Zhang <jizha@amd.com> Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com> Co-authored-by: illsilin <Illia.Silin@amd.com>
33 lines
983 B
C++
33 lines
983 B
C++
|
|
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#include "ck/host/device_gemm_multiple_d/problem.hpp"
|
|
#include "ck/host/device_gemm_multiple_d/operation.hpp"
|
|
#include "ck/host/utils.hpp"
|
|
#include <algorithm>
|
|
|
|
namespace ck {
|
|
namespace host {
|
|
namespace device_gemm_multiple_d {
|
|
|
|
std::string Problem::GetIncludeHeader() const
|
|
{
|
|
return "ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_xdl_cshuffle.hpp";
|
|
}
|
|
|
|
std::vector<Solution> Problem::GetSolutions(const std::string& arch) const
|
|
{
|
|
if(get_xdlop_archs().count(arch) == 0)
|
|
return {};
|
|
auto ops = ck::host::device_gemm_multiple_d::Operation_Xdl_CShuffle::CreateOperations(*this);
|
|
std::vector<Solution> result;
|
|
std::transform(ops.begin(), ops.end(), std::back_inserter(result), [&](const auto& op) {
|
|
return op.ToSolution();
|
|
});
|
|
return result;
|
|
}
|
|
|
|
} // namespace device_gemm_multiple_d
|
|
} // namespace host
|
|
} // namespace ck
|