"vscode:/vscode.git/clone" did not exist on "e17b495dff05a641e6a15efdb92b603b2f111a8a"
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
...@@ -255,20 +255,11 @@ float gemm_calc(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& ...@@ -255,20 +255,11 @@ float gemm_calc(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config&
} }
else else
{ {
// Tail number always Full - #PrefetchStages std::ostringstream err;
if(tail_num == ck_tile::TailNumber::Full) err << "Num K loop must be larger than number of prefetech stages."
{ << "\n PrefetchStages: " << BaseGemmPipeline::PrefetchStages << "\n File: " << __FILE__
Run(ck_tile::bool_constant<false>{}, << ":" << __LINE__ << ", in function: " << __func__;
ck_tile::integral_constant<ck_tile::TailNumber, ck_tile::TailNumber::Full>{}); throw std::runtime_error(err.str());
}
else
{
std::ostringstream err;
err << "When there's no hot loop, this tail number \"" << tail_num
<< "\" is not supported! PrefetchStages: " << BaseGemmPipeline::PrefetchStages
<< "\n File: " << __FILE__ << ":" << __LINE__ << ", in function: " << __func__;
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