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
cc9dd3b6
Commit
cc9dd3b6
authored
Jul 05, 2024
by
Harisankar Sadasivan
Browse files
print messages for arguments
parent
22329520
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
8 deletions
+8
-8
example/01_gemm/README.md
example/01_gemm/README.md
+1
-1
example/01_gemm/common.hpp
example/01_gemm/common.hpp
+7
-7
No files found.
example/01_gemm/README.md
View file @
cc9dd3b6
...
@@ -16,7 +16,7 @@ arg1: verification (0=no, 1=yes)
...
@@ -16,7 +16,7 @@ arg1: verification (0=no, 1=yes)
arg2: initialization
(
0
=
no init,
1
=
integer value,
2
=
decimal value
)
arg2: initialization
(
0
=
no init,
1
=
integer value,
2
=
decimal value
)
arg3:
time
kernel
(
0
=
no,
1
=
yes
)
arg3:
time
kernel
(
0
=
no,
1
=
yes
)
arg4 to 9: M
(
256x
)
, N
(
128x
)
, K
(
32x
)
, StrideA, StrideB, StrideC
arg4 to 9: M
(
256x
)
, N
(
128x
)
, K
(
32x
)
, StrideA, StrideB, StrideC
arg10: stream-k
select
(
0: all DP, 1: 1-tile SK, 2: 2-tile SK
)
arg10: stream-k
select
(
-1
: default config,
0: all DP, 1: 1-tile SK, 2: 2-tile SK
)
arg11: Grid_size
(
-1
for
max occupancy
)
arg11: Grid_size
(
-1
for
max occupancy
)
bin/example_gemm_xdl_fp16_streamk_v3 1 2 1 3840 4096 4096 4096 4096 4096 1
-1
bin/example_gemm_xdl_fp16_streamk_v3 1 2 1 3840 4096 4096 4096 4096 4096 1
-1
a_m_k: dim 2, lengths
{
3840, 4096
}
, strides
{
4096, 1
}
a_m_k: dim 2, lengths
{
3840, 4096
}
, strides
{
4096, 1
}
...
...
example/01_gemm/common.hpp
100644 → 100755
View file @
cc9dd3b6
...
@@ -174,13 +174,13 @@ bool parse_cmd_args<ProblemSizeStreamK_universal>(int argc,
...
@@ -174,13 +174,13 @@ bool parse_cmd_args<ProblemSizeStreamK_universal>(int argc,
}
}
else
else
{
{
std
::
cerr
<<
"arg1: verification (0=no, 1=yes)"
<<
std
::
endl
std
::
cerr
<<
"arg
2
:
initializ
ation (0=no
init, 1=integer value, 2=decimal value)"
<<
"arg
1
:
verific
ation (0=no
, 1=yes)"
<<
std
::
endl
<<
std
::
endl
<<
"arg2: initialization (0=no init, 1=integer value, 2=decimal value)"
<<
std
::
endl
<<
"arg3: time kernel (0=no, 1=yes)"
<<
std
::
endl
<<
"arg3: time kernel (0=no, 1=yes)"
<<
std
::
endl
<<
"arg4 to 9: M (256x), N(128x), K(32x), StrideA, StrideB, StrideC"
<<
std
::
endl
<<
"arg4 to 9: M (256x), N(128x), K(32x), StrideA, StrideB, StrideC"
<<
std
::
endl
<<
"arg10: stream-k select (0: all DP, 1: 1-tile SK, 2: 2-tile SK)"
<<
"arg10: stream-k select (
-1: default config,
0: all DP, 1: 1-tile SK, 2: 2-tile SK)"
<<
"
\n
arg11: Grid_size(-1 for max occupancy)"
<<
std
::
endl
;
<<
"
\n
arg11: Grid_size(-1 for max occupancy)"
<<
std
::
endl
;
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