Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
6b6fcd37
"vscode:/vscode.git/clone" did not exist on "e17b495dff05a641e6a15efdb92b603b2f111a8a"
Unverified
Commit
6b6fcd37
authored
Feb 14, 2025
by
Bartłomiej Kocot
Committed by
GitHub
Feb 14, 2025
Browse files
[CK TILE] Check for num loop < Prefetch Stages in gemm (#1886)
parent
4cfb24fe
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
14 deletions
+5
-14
example/ck_tile/03_gemm/universal_gemm.cpp
example/ck_tile/03_gemm/universal_gemm.cpp
+5
-14
No files found.
example/ck_tile/03_gemm/universal_gemm.cpp
View file @
6b6fcd37
...
@@ -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
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment