ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move another function to cpp file

Browse Source
This commit is contained in:
Paul
2023-05-25 15:56:40 -05:00
parent dd6fd8bb62
commit 856419e802
2 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
library/src/jit_library/include/ck/host/device_gemm_multiple_d.hpp
View File

@@ -63,27 +63,9 @@ private:
;
public:
auto GetHeaders() const
{
return ck_headers();
}
std::string GetIncludeHeader() const;
std::string GetIncludeHeader() const
{
return instance::gemm_add_add_fastgelu_instances{}.get_include_header();
}
std::vector<Solution> GetSolutions(const std::string& arch) const
{
std::vector<Solution> solutions;
const auto num_instances = GetInstances(arch).size();
for (auto i = 0; i < num_instances; ++i)
{
solutions.push_back(MakeSolution(i, arch));
}
return solutions;
}
std::vector<Solution> GetSolutions(const std::string& arch) const;
};
} // namespace device_gemm_multiple_d

17
library/src/jit_library/src/device_gemm_multiple_d.cpp
View File

@@ -136,6 +136,23 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const
return Solution{str, block_size, grid_size};
}
std::string Problem::GetIncludeHeader() const
{
return instance::gemm_add_add_fastgelu_instances{}.get_include_header();
}
std::vector<Solution> Problem::GetSolutions(const std::string& arch) const
{
std::vector<Solution> solutions;
const auto num_instances = GetInstances(arch).size();
for (auto i = 0; i < num_instances; ++i)
{
solutions.push_back(MakeSolution(i, arch));
}
return solutions;
}
} // namespace device_gemm_multiple_d
} // namespace host