Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "cd51758fade4119b3f6233444c3bfac91ed5eba9"
Unverified Commit 6b6fcd37 authored by Bartłomiej Kocot's avatar Bartłomiej Kocot Committed by GitHub
Browse files

[CK TILE] Check for num loop < Prefetch Stages in gemm (#1886)

parent 4cfb24fe
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 14 deletions
example/ck_tile/03_gemm/universal_gemm.cpp
View file @ 6b6fcd37
...@@ -254,22 +254,13 @@ float gemm_calc(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& ...@@ -254,22 +254,13 @@ float gemm_calc(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config&
#endif #endif
} }
else else
{
// Tail number always Full - #PrefetchStages
if(tail_num == ck_tile::TailNumber::Full)
{
Run(ck_tile::bool_constant<false>{},
ck_tile::integral_constant<ck_tile::TailNumber, ck_tile::TailNumber::Full>{});
}
else
{ {
std::ostringstream err; std::ostringstream err;
err << "When there's no hot loop, this tail number \"" << tail_num err << "Num K loop must be larger than number of prefetech stages."
<< "\" is not supported! PrefetchStages: " << BaseGemmPipeline::PrefetchStages << "\n PrefetchStages: " << BaseGemmPipeline::PrefetchStages << "\n File: " << __FILE__
<< "\n File: " << __FILE__ << ":" << __LINE__ << ", in function: " << __func__; << ":" << __LINE__ << ", in function: " << __func__;
throw std::runtime_error(err.str()); throw std::runtime_error(err.str());
} }
}
return ave_time; return ave_time;
} }
......
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