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
Commits
346e837f
Commit
346e837f
authored
Jun 08, 2022
by
ltqin
Browse files
fix verify data(mulitk0)
parent
179c561e
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
example/01_gemm/gemm_xdl_skip_b_lds_fp16.cpp
example/01_gemm/gemm_xdl_skip_b_lds_fp16.cpp
+2
-2
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_skip_b_lds_v1.hpp
...operation/gpu/grid/gridwise_gemm_xdlops_skip_b_lds_v1.hpp
+1
-1
No files found.
example/01_gemm/gemm_xdl_skip_b_lds_fp16.cpp
View file @
346e837f
...
@@ -87,10 +87,10 @@ template <typename DataType>
...
@@ -87,10 +87,10 @@ template <typename DataType>
std
::
ostream
&
show_2d_matrix
(
std
::
ostream
&
os
,
Tensor
<
DataType
>&
matrix
)
std
::
ostream
&
show_2d_matrix
(
std
::
ostream
&
os
,
Tensor
<
DataType
>&
matrix
)
{
{
os
<<
"["
<<
std
::
endl
;
os
<<
"["
<<
std
::
endl
;
for
(
in
t
x
=
0
;
x
<
matrix
.
mDesc
.
GetLengths
()[
0
];
x
++
)
for
(
size_
t
x
=
0
;
x
<
matrix
.
mDesc
.
GetLengths
()[
0
];
x
++
)
{
{
os
<<
"["
;
os
<<
"["
;
for
(
in
t
y
=
0
;
y
<
matrix
.
mDesc
.
GetLengths
()[
1
];
y
++
)
for
(
size_
t
y
=
0
;
y
<
matrix
.
mDesc
.
GetLengths
()[
1
];
y
++
)
{
{
os
<<
std
::
setw
(
5
)
<<
static_cast
<
float
>
(
matrix
(
x
,
y
));
os
<<
std
::
setw
(
5
)
<<
static_cast
<
float
>
(
matrix
(
x
,
y
));
}
}
...
...
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_skip_b_lds_v1.hpp
View file @
346e837f
...
@@ -191,7 +191,7 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_skip_b_lds_v1
...
@@ -191,7 +191,7 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_skip_b_lds_v1
// 2-stage prefetch currently only support even number of K0 loop
// 2-stage prefetch currently only support even number of K0 loop
// TODO: add support for odd number of K0 loop
// TODO: add support for odd number of K0 loop
if
(
!
((
K0
/
K0PerBlock
)
%
2
==
0
))
if
(
!
((
K0
/
K0PerBlock
)
%
MultiK0
==
0
))
{
{
return
false
;
return
false
;
}
}
...
...
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