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
f7f9954d
Commit
f7f9954d
authored
Jun 25, 2024
by
Harisankar Sadasivan
Browse files
removed conflicts in struct members between streamk and unievrsal streamk
parent
b469ec5f
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
15 additions
and
3 deletions
+15
-3
example/01_gemm/common.hpp
example/01_gemm/common.hpp
+12
-0
example/01_gemm/gemm_xdl_fp16_streamk_v3.cpp
example/01_gemm/gemm_xdl_fp16_streamk_v3.cpp
+1
-1
example/01_gemm/run_gemm_example_streamk_v2.inc
example/01_gemm/run_gemm_example_streamk_v2.inc
+2
-2
No files found.
example/01_gemm/common.hpp
View file @
f7f9954d
...
...
@@ -43,6 +43,18 @@ struct ProblemSizeStreamK final
ck
::
index_t
StrideB
=
4096
;
ck
::
index_t
StrideC
=
4096
;
ck
::
index_t
NumSKBlocks
=
-
1
;
};
struct
ProblemSizeStreamK_universal
final
{
ck
::
index_t
M
=
3840
;
ck
::
index_t
N
=
4096
;
ck
::
index_t
K
=
4096
;
ck
::
index_t
StrideA
=
4096
;
ck
::
index_t
StrideB
=
4096
;
ck
::
index_t
StrideC
=
4096
;
ck
::
index_t
Grid_size
=
-
1
;
// defaults to max occupancy
ck
::
index_t
Streamk_sel
=
1
;
// defaults to 1-tile SK
};
...
...
example/01_gemm/gemm_xdl_fp16_streamk_v3.cpp
View file @
f7f9954d
...
...
@@ -45,4 +45,4 @@ using ReferenceGemmInstance = ck::tensor_operation::host::
#include "run_gemm_example_streamk_v2.inc"
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_
split
k_example
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_
universal_stream
k_example
(
argc
,
argv
);
}
example/01_gemm/run_gemm_example_streamk_v2.inc
View file @
f7f9954d
...
...
@@ -290,9 +290,9 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
return
pass
;
}
bool
run_gemm_
split
k_example
(
int
argc
,
char
*
argv
[])
bool
run_gemm_
universal_stream
k_example
(
int
argc
,
char
*
argv
[])
{
ProblemSizeStreamK
problem_size
;
ProblemSizeStreamK
_universal
problem_size
;
ExecutionConfig
config
;
return
!
parse_cmd_args
(
argc
,
argv
,
problem_size
,
config
)
||
run_gemm
(
problem_size
,
config
);
...
...
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