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
MIGraphX
Commits
d2025171
Commit
d2025171
authored
Aug 21, 2023
by
Paul
Browse files
Show min/max times
parent
24d9ecbe
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
1 deletion
+3
-1
src/targets/gpu/compile_ops.cpp
src/targets/gpu/compile_ops.cpp
+2
-0
src/targets/gpu/fuse_mlir.cpp
src/targets/gpu/fuse_mlir.cpp
+1
-1
No files found.
src/targets/gpu/compile_ops.cpp
View file @
d2025171
...
@@ -190,6 +190,8 @@ struct compile_plan
...
@@ -190,6 +190,8 @@ struct compile_plan
});
});
auto
i
=
std
::
distance
(
times
.
begin
(),
std
::
min_element
(
times
.
begin
(),
times
.
end
()));
auto
i
=
std
::
distance
(
times
.
begin
(),
std
::
min_element
(
times
.
begin
(),
times
.
end
()));
std
::
cout
<<
"Fastest solution: "
<<
config
->
solutions
.
at
(
i
)
<<
std
::
endl
;
std
::
cout
<<
"Fastest solution: "
<<
config
->
solutions
.
at
(
i
)
<<
std
::
endl
;
std
::
cout
<<
"Fastest time: "
<<
*
std
::
min_element
(
times
.
begin
(),
times
.
end
())
<<
std
::
endl
;
std
::
cout
<<
"Slowest time: "
<<
*
std
::
max_element
(
times
.
begin
(),
times
.
end
())
<<
std
::
endl
;
pc
.
insert
(
preop
.
name
(),
config
->
problem
,
config
->
solutions
.
at
(
i
));
pc
.
insert
(
preop
.
name
(),
config
->
problem
,
config
->
solutions
.
at
(
i
));
if
(
not
results
[
i
].
has_value
())
if
(
not
results
[
i
].
has_value
())
MIGRAPHX_THROW
(
"No valid tuned compilation."
);
MIGRAPHX_THROW
(
"No valid tuned compilation."
);
...
...
src/targets/gpu/fuse_mlir.cpp
View file @
d2025171
...
@@ -237,7 +237,7 @@ struct find_mlir_op
...
@@ -237,7 +237,7 @@ struct find_mlir_op
"recip"
,
"recip"
,
"rsqrt"
,
"rsqrt"
,
// There are bugs in MLIR right now for models using sigmoid so disable it for now
// There are bugs in MLIR right now for models using sigmoid so disable it for now
//
"sigmoid",
"sigmoid"
,
"softmax"
,
"softmax"
,
"tanh"
,
"tanh"
,
};
};
...
...
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