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
c3952566
"...resnet50_tensorflow.git" did not exist on "2fe714954d32eafac9be59e178cfea3e2e258935"
Commit
c3952566
authored
Mar 14, 2022
by
Jing Zhang
Browse files
improve code
parent
892760bd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
8 deletions
+9
-8
profiler/src/profile_grouped_gemm.cpp
profiler/src/profile_grouped_gemm.cpp
+9
-8
No files found.
profiler/src/profile_grouped_gemm.cpp
View file @
c3952566
...
...
@@ -26,7 +26,7 @@ enum GemmDataType
INT8_INT8_INT8
,
// 3
};
std
::
vector
<
int
>
stringTo
Array
(
char
*
input
)
std
::
vector
<
int
>
argToInt
Array
(
char
*
input
)
{
std
::
vector
<
int
>
out
;
...
...
@@ -56,7 +56,8 @@ int profile_grouped_gemm(int argc, char* argv[])
printf
(
"arg5: initialization (0: no init; 1: integer value; 2: decimal value)
\n
"
);
printf
(
"arg8: print tensor value (0: no; 1: yes)
\n
"
);
printf
(
"arg7: run kernel # of times (>1)
\n
"
);
printf
(
"arg8 to 13: Ms, Ns, Ks, StrideAs, StrideBs, StrideCs
\n
"
);
printf
(
"arg8 to 13: Ms, Ns, Ks, StrideAs, StrideBs, StrideCs (e.g., 256,256 128,128 64,64 "
"64,64 64,64 128,128)
\n
"
);
exit
(
1
);
}
...
...
@@ -67,13 +68,13 @@ int profile_grouped_gemm(int argc, char* argv[])
const
bool
do_log
=
std
::
stoi
(
argv
[
6
]);
const
int
nrepeat
=
std
::
stoi
(
argv
[
7
]);
const
auto
Ms
=
stringTo
Array
(
argv
[
8
]);
const
auto
Ns
=
stringTo
Array
(
argv
[
9
]);
const
auto
Ks
=
stringTo
Array
(
argv
[
10
]);
const
auto
Ms
=
argToInt
Array
(
argv
[
8
]);
const
auto
Ns
=
argToInt
Array
(
argv
[
9
]);
const
auto
Ks
=
argToInt
Array
(
argv
[
10
]);
const
auto
StrideAs
=
stringTo
Array
(
argv
[
11
]);
const
auto
StrideBs
=
stringTo
Array
(
argv
[
12
]);
const
auto
StrideCs
=
stringTo
Array
(
argv
[
13
]);
const
auto
StrideAs
=
argToInt
Array
(
argv
[
11
]);
const
auto
StrideBs
=
argToInt
Array
(
argv
[
12
]);
const
auto
StrideCs
=
argToInt
Array
(
argv
[
13
]);
if
(
data_type
==
GemmDataType
::
F16_F16_F16
&&
layout
==
GemmMatrixLayout
::
MK_KN_MN
)
{
...
...
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