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
c345719a
Commit
c345719a
authored
Dec 05, 2021
by
Chao Liu
Browse files
refactor
parent
1f8e8231
Changes
21
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
6 deletions
+13
-6
profiler/profiler.cpp
profiler/profiler.cpp
+13
-6
No files found.
profiler/profiler.cpp
View file @
c345719a
...
@@ -5,22 +5,29 @@
...
@@ -5,22 +5,29 @@
#include <stdlib.h>
#include <stdlib.h>
#include <half.hpp>
#include <half.hpp>
int
gemm_profiler
(
int
,
char
*
[]);
int
profile_gemm
(
int
,
char
*
[]);
int
conv_profiler
(
int
,
char
*
[]);
int
profile_conv_fwd
(
int
,
char
*
[]);
// int profile_conv_fwd_bias_relu_add(int, char*[]);
int
main
(
int
argc
,
char
*
argv
[])
int
main
(
int
argc
,
char
*
argv
[])
{
{
if
(
strcmp
(
argv
[
1
],
"gemm"
)
==
0
)
if
(
strcmp
(
argv
[
1
],
"gemm"
)
==
0
)
{
{
return
gemm_
profile
r
(
argc
,
argv
);
return
profile
_gemm
(
argc
,
argv
);
}
}
else
if
(
strcmp
(
argv
[
1
],
"conv"
)
==
0
)
else
if
(
strcmp
(
argv
[
1
],
"conv
_fwd
"
)
==
0
)
{
{
return
conv_
profile
r
(
argc
,
argv
);
return
profile
_conv_fwd
(
argc
,
argv
);
}
}
#if 0
else if(strcmp(argv[1], "conv_fwd_bias_relu_add") == 0)
{
return profile_conv_fwd_bias_relu_add(argc, argv);
}
#endif
else
else
{
{
printf
(
"arg1: tensor operation (
gemm=GEMM, conv=
Convolution)
\n
"
);
printf
(
"arg1: tensor operation (
conv_fwd: Forward
Convolution)
\n
"
);
return
0
;
return
0
;
}
}
}
}
Prev
1
2
Next
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