Commit 000c8bcf authored by Umang Yadav's avatar Umang Yadav
Browse files

Merge branch 'migx-jit-lib' into migraphx

parents 11cab2d5 8f9c0243
...@@ -577,9 +577,7 @@ struct DeviceGemmMultipleD_Xdl_CShuffle : public DeviceGemmMultipleD<ALayout, ...@@ -577,9 +577,7 @@ struct DeviceGemmMultipleD_Xdl_CShuffle : public DeviceGemmMultipleD<ALayout,
static bool IsSupportedArgument(const Argument& arg) static bool IsSupportedArgument(const Argument& arg)
{ {
if(!(ck::get_device_name() == "gfx908" || ck::get_device_name() == "gfx90a" || if(!ck::is_xdl_supported())
ck::get_device_name() == "gfx940" || ck::get_device_name() == "gfx941" ||
ck::get_device_name() == "gfx942"))
{ {
return false; return false;
} }
......
...@@ -141,6 +141,8 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const ...@@ -141,6 +141,8 @@ Solution Problem::MakeSolution(std::size_t idx, const std::string& arch) const
std::string{}, std::string{},
[](const std::string& a, const std::string& b) { return a.empty() ? b : a + ", " + b; }); [](const std::string& a, const std::string& b) { return a.empty() ? b : a + ", " + b; });
str = params.front() + "< " + str + ">"; str = params.front() + "< " + str + ">";
if (params.back().find("v2") != std::string::npos and K % k_per_block != 0)
str = "";
return Solution{str, block_size, grid_size}; return Solution{str, block_size, grid_size};
} }
...@@ -156,7 +158,9 @@ std::vector<Solution> Problem::GetSolutions(const std::string& arch) const ...@@ -156,7 +158,9 @@ std::vector<Solution> Problem::GetSolutions(const std::string& arch) const
const std::size_t num_instances = GetInstances(arch).size(); const std::size_t num_instances = GetInstances(arch).size();
for(std::size_t i = 0; i < num_instances; ++i) for(std::size_t i = 0; i < num_instances; ++i)
{ {
solutions.push_back(MakeSolution(i, arch)); auto solution = MakeSolution(i, arch);
if (solution.template_str != "")
solutions.push_back(solution);
} }
return solutions; return solutions;
......
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