Commit 856419e8 authored by Paul's avatar Paul
Browse files

Move another function to cpp file

parent dd6fd8bb
...@@ -63,27 +63,9 @@ private: ...@@ -63,27 +63,9 @@ private:
; ;
public: public:
auto GetHeaders() const std::string GetIncludeHeader() const;
{
return ck_headers();
}
std::string GetIncludeHeader() const std::vector<Solution> GetSolutions(const std::string& arch) 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;
}
}; };
} // namespace device_gemm_multiple_d } // namespace device_gemm_multiple_d
......
...@@ -136,6 +136,23 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const ...@@ -136,6 +136,23 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const
return Solution{str, block_size, grid_size}; 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 device_gemm_multiple_d
} // namespace host } // namespace host
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment