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
c6dcf20d
"git@developer.sourcefind.cn:modelzoo/alphafold2_jax.git" did not exist on "ed0886cb9806eaa1d3c8a54a339c98fdd850ccf3"
Commit
c6dcf20d
authored
Jan 21, 2025
by
Adam Osewski
Browse files
Fix static assert.
parent
465f8e6a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
10 deletions
+10
-10
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v3.hpp
...tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v3.hpp
+10
-10
No files found.
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v3.hpp
View file @
c6dcf20d
...
...
@@ -249,19 +249,19 @@ struct GemmPipelineAgBgCrCompV3 : public BaseGemmPipelineAgBgCrCompV3<Problem>
std
::
is_same_v
<
ALayout
,
tensor_layout
::
gemm
::
ColumnMajor
>
;
constexpr
bool
is_b_row_major
=
std
::
is_same_v
<
BLayout
,
tensor_layout
::
gemm
::
RowMajor
>
;
static_assert
(
(
is_a_col_major
?
(
M
PerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
1
{}]
K
PerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
0
{}])
&&
static_assert
(
is_a_col_major
?
(
K
PerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
0
{}]
&&
M
PerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
1
{}])
:
(
MPerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I0
{}]
KPerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
0
{}]),
(
MPerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I0
{}]
&&
KPerBlock
==
ADramBlockWindowTmp
{}.
get_window_lengths
()[
I
1
{}]),
"A block window has incorrect lengths for defined ALayout!"
);
static_assert
(
(
is_b_row_major
?
(
KPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I0
{}]
NPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I1
{}])
&&
static_assert
(
is_b_row_major
?
(
KPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I0
{}]
&&
NPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I1
{}])
:
(
NPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I
1
{}]
KPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I
0
{}]),
(
NPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I
0
{}]
&&
KPerBlock
==
BDramBlockWindowTmp
{}.
get_window_lengths
()[
I
1
{}]),
"B block window has incorrect lengths for defined BLayout!"
);
// ------------------------------------------------------------------------------------
...
...
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