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
MIGraphX
Commits
f903e858
Commit
f903e858
authored
Oct 20, 2022
by
Paul
Browse files
Format
parent
7c77c682
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm_includes.hpp
...gpu/kernels/include/migraphx/kernels/ck_gemm_includes.hpp
+6
-5
No files found.
src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm_includes.hpp
View file @
f903e858
...
...
@@ -228,10 +228,10 @@ struct CK_DeviceGemmMultipleD
typename
EGridDesc_M_N
,
typename
Block2ETileMap
>
static
constexpr
bool
CheckValidity
(
const
AGridDesc_M_K
&
a_grid_desc_m_k
,
const
BGridDesc_N_K
&
b_grid_desc_n_k
,
const
DsGridDesc_M_N
&
ds_grid_desc_m_n
,
const
EGridDesc_M_N
&
e_grid_desc_m_n
,
const
Block2ETileMap
&
block_2_etile_map
)
const
BGridDesc_N_K
&
b_grid_desc_n_k
,
const
DsGridDesc_M_N
&
ds_grid_desc_m_n
,
const
EGridDesc_M_N
&
e_grid_desc_m_n
,
const
Block2ETileMap
&
block_2_etile_map
)
{
constexpr
auto
M
=
a_grid_desc_m_k
.
GetLength
(
I0
);
constexpr
auto
N
=
b_grid_desc_n_k
.
GetLength
(
I0
);
...
...
@@ -246,7 +246,8 @@ struct CK_DeviceGemmMultipleD
// check block-to-E-tile
static_assert
(
block_2_etile_map
.
CheckValidity
(
e_grid_desc_m_n
));
return
GridwiseGemm
::
CheckValidity
(
a_grid_desc_m_k
,
b_grid_desc_n_k
,
ds_grid_desc_m_n
,
e_grid_desc_m_n
,
block_2_etile_map
);
return
GridwiseGemm
::
CheckValidity
(
a_grid_desc_m_k
,
b_grid_desc_n_k
,
ds_grid_desc_m_n
,
e_grid_desc_m_n
,
block_2_etile_map
);
}
AElementwiseOperation
a_element_op
{};
...
...
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